2023.7.12 mean, expand【torch】

meanはテンソルに対して指定した方向の平均を求める関数、int型には対応しておらず、元データはfloat型とする必要がある。

code:python

import torch

print('# 1Dテンソル')

a = torch.tensor(1, 2, 2, 2.5).float()

print(a.mean(0))

print('# 4×1テンソル')

b = torch.tensor(1], 2, 2, [2.5).float()

print(b.mean(0))

print(b.mean(1))

print('# 2×4テンソル')

c = torch.tensor(1,2,2,3], [2,1,3,2).float()

print(c.mean(0))

print(c.mean(1))

テンソルを拡張させるexpand、拡張された要素は同じオブジェクトを共有していることに注意。

code:python

import torch

a = torch.tensor(0)

a2 = a.expand(3)

print(a2)

b = torch.tensor(0, 1, 2, 3)

b2 = b.expand(3,-1)

print(b2)

# b3 = b.expand(-1, 3)

# print(b3)

b21,2 = 100

print(b2) # 共有している