【Pytorch】基本的な行列を作る（宣言）する方法【Python】

# listを基にしたTensor作成
a = torch.tensor([[1, 2], [3, 4]])
print("torch.tensor:")
print(a)
print()

# 出力
# torch.tensor:
# tensor([[1, 2],
#         [3, 4]])

# 全要素が1の行列 (numpy.ones)
a = torch.ones((2, 3))
print("torch.ones:")
print(a)
print()

# 出力
# torch.tensor:
# tensor([[1, 2],
#         [3, 4]])

# 全要素が1の行列 (numpy.ones)
a = torch.ones((2, 3))
print("torch.ones:")
print(a)
print()

# 出力
# torch.ones:
tensor([[1., 1., 1.],
        [1., 1., 1.]])

# 全要素が0の行列 (numpy.ones)
a = torch.zeros((2, 3))
print("torch.zeros:")
print(a)
print()

# 出力
# torch.zeros:
# tensor([[0., 0., 0.],
#         [0., 0., 0.]])

# 指定した値で満たされた行列 (numpy.full)
a = torch.full((2, 3), fill_value=99)
print("torch.full:")
print(a)
print()

# 出力
# torch.full:
# tensor([[99, 99, 99],
#         [99, 99, 99]])

# 単位行列 (numpy.eye)
a = torch.eye(2)
print("torch.eye:")
print(a)

# 出力
# torch.eye:
# tensor([[1., 0.],
#         [0., 1.]])