【Pytorch】テンソルのスカラー演算まとめ
Pytorchで定義したテンソルのスカラー演算はnumpyと同様に実装することができる。
演算で使うテンソルa，bを以下で宣言する。
a = torch.rand((2, 3))
b = torch.rand((2, 3))
# a
# tensor([[0.5573, 0.2211, 0.7223],
# [0.9716, 0.1025, 0.0126]])
# b
# tensor([[0.4291, 0.7196, 0.3006],
# [0.5679, 0.5095, 0.1978]])足し算
a+bをしたいときは以下のコードになる。
printで結果を出力している。
print(a + b)
# 出力
# tensor([[0.9865, 0.9407, 1.0229],
# [1.5395, 0.6120, 0.2104]])引き算
a－bをしたいときは以下のコードになる。
print(a - b)
# 出力
# tensor([[ 0.1282, -0.4985, 0.4216],
# [ 0.4037, -0.4071, -0.1851]])掛け算
a×bをしたいときは以下のコードになる。
print(a * b)
# 出力
# tensor([[0.2392, 0.1591, 0.2171],
# [0.5517, 0.0522, 0.0025]])割り算
a÷bをしたいときは以下のコードになる。
print(a / b)
# 出力
# tensor([[1.2987, 0.3072, 2.4026],
# [1.7110, 0.2011, 0.0639]])log
log(a)を求めたいときは以下のコードになる。
print(torch.log(a))
# 出力
# tensor([[-0.5846, -1.5092, -0.3254],
# [-0.0288, -2.2782, -4.3705]])exp
exp(a)を求めたいときは以下のコードになる。
print(torch.exp(a))
# 出力
# tensor([[1.7460, 1.2474, 2.0591],
# [2.6422, 1.1079, 1.0127]])ルート（sqrt）
aのルートを求めたいときは以下のコードになる。
print(torch.sqrt(a))
# 出力
# tensor([[0.7465, 0.4702, 0.8499],
# [0.9857, 0.3201, 0.1125]])以上がスカラー演算のまとめ。
人気記事
人気記事はこちら。
