例えば
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.
pytorchのdimはnumpyのaxisと同じ
Last updated at Posted at 2017-09-19
Register as a new user and use Qiita more conveniently
- You get articles that match your needs
- You can efficiently read back useful information
- You can use dark theme