pytorch单个张量的的函数运算

pytorch单个张量的的函数运算操作既可以由张量自带的方法实现，比如下面的a.sqrt()，也可以由torch包中的一些函数实现：

import torch

a = torch.rand(3, 4)  # 产生一个3*4的张量
print(a)

print(a.sqrt())  # 张量的平方根，张量的内部方法,a不会随之改变

print(torch.sqrt(a))  # 张量的函数形式，张量的内部方法,a不会随之改变

print(a.sqrt_())  # 平方根原地操作，a会随之改变
print(a)

print(torch.sum(a))  # 默认对所有的元素求和

print(torch.sum(a, 0))  # 对第0维元素求和,实际就是对每一列求和

print(torch.sum(a, 1))  # 对第1维元素求和,实际就是对每一行求和

print(a.mean())  # 对所有元素求平均

print(a.mean(0))  # 对第0维元素求平均，实际就是对每一列求平均

print(a.mean(1))  # 对第1维元素求平均，实际就是对每一行求平均

Done!!!