【Pytorch】テンソルをnumpy・list・scalarに変換する方法

Pytorchで定義されたテンソルをNumpy配列、list、scalarに変換する方法をまとめておく。

元の配列として以下を用いる。

a = torch.ones((2, 3))
print(a)

# 出力
# tensor([[1., 1., 1.],
#         [1., 1., 1.]])

Numpy配列への変換は以下のコードのように行う。

.numpy()を使う。

# numpy.ndarrayへの変換
print("a.numpy():")
print(a.numpy())

# 出力
# a.numpy():
# [[1. 1. 1.]
#  [1. 1. 1.]]

listへの変換は以下のコードのように行う。

.tolist()を使う。

# listへの変換
print("a.tolist():")
print(a.tolist())

# 出力
# a.tolist():
# [[1.0, 1.0, 1.0], [1.0, 1.0, 1.0]]

scalarへの変換は以下のコードのように行う。

.item()を使う。

# scalarへの変換には.item()を使う
print("a.item():")
print(a.sum().item())

# 出力
# a.item():
# 6.0