【Pytorch】テンソルのサイズを確認する方法(size・shape)【Tensor】

【Pytorch】テンソルのサイズを確認する方法(size・shape)

PytorchではTensor（テンソル）のサイズを確認するための方法として２種類ある。

.size()と.shapeで調べることが出来る。

これは.size()のエイリアスとして.shapeが存在しているということらしい。

.shapeはnumpyでも同じようにサイズを確認できる。

以下のようにコードを書いて確認する。

a = torch.ones((3, 4))

print(a.size())  # Tensorのサイズを取得
print(a.shape)  # .shapeでも可能

print(a.size(0))  # Tensorの0次元目のサイズを取得

# 出力
# torch.Size([3, 4])
# torch.Size([3, 4])
# 3

以上がテンソルのサイズを確認する方法。