【PyTorch学习7】运算函数

• mean/sum/pow/sqrt/exp/log

data = torch.tensor([[1.,2.,3.],[4.,5.,6.]])
# mean 默认所有数据计算均值
data1 = data.mean()  # tensor(3.5000)
# mean 计算0维度的均值（列）
data2 = data.mean(0)  # tensor([2.5000, 3.5000, 4.5000])
# mean 计算1维度的均值（行）
data3 = data.mean(1)  # tensor([2., 5.])

# sum 默认计算所有数据的和
data4 = data.sum()  # tensor(21.)
# sum 计算0维度的和（列）
data5 = data.sum(0)  # tensor([5., 7., 9.])
# sum 计算1维度的和（行）
data6 = data.sum(1)  # tensor([ 6., 15.])

# pow 计算每个数据的n次方
data7 = data.pow(2)  # tensor([[ 1.,  4.,  9.],[16., 25., 36.]])

# sqrt 计算每个数据的平方根
data8 = data.sqrt()  # tensor([[1.0000, 1.4142, 1.7321],[2.0000, 2.2361, 2.4495]])

# exp 计算每个数据e的n次方
data9 = data.exp()  # tensor([[  2.7183,   7.3891,  20.0855],[ 54.5981, 148.4132, 403.4288]])

# log 计算对数(默认以e为底)
data10 = data.log()  # tensor([[0.0000, 0.6931, 1.0986],[1.3863, 1.6094, 1.7918]])
# log 以2为底
data11 = data.log2()  # tensor([[0.0000, 1.0000, 1.5850],[2.0000, 2.3219, 2.5850]])