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

スポンサーリンク

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

Pytorchを使って行列を作る(宣言)する方法をまとめておく。

ここではサンプルプログラムを紹介する。

基本的な行列を作成するコード

listを基にしたTensorの作成例↓

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

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

numpyを基にしたTensorの作成例↓

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

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

全要素が1の行列 (numpy.ones)の作成例↓

# 全要素が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)の作成例↓

# 全要素が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)の作成例↓

# 指定した値で満たされた行列 (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)の作成例↓

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

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

以上がPytorchを用いた基本的な行列の作成方法。

スポンサーリンク

人気記事

人気記事はこちら。

CUDA、cuDNNのバージョンをターミナルで調べるコマンド
【Pytorch】テンソルを連結する方法(cat・stack)
【Pytorch】テンソルの次元を追加・削除する方法【dim】
【Protobuf】"TypeError: Descriptors cannot not be created directly."を解決する【solved】
【Python】Tensorflowをダウングレード・アップグレードするコマンド
タイトルとURLをコピーしました