例えば
torch.max(input, dim, ...)
torch.squeeze(input, dim,...)
のdim
引数は、numpyの
np.max(a, axis, ...)
np.squeeze(a, axis, ...)
axis
引数にあたる。
More than 5 years have passed since last update.
例えば
torch.max(input, dim, ...)
torch.squeeze(input, dim,...)
のdim
引数は、numpyの
np.max(a, axis, ...)
np.squeeze(a, axis, ...)
axis
引数にあたる。
Register as a new user and use Qiita more conveniently