PyTorch 为每个张量封装很多实用的计算函数： 均值 求和 平方 平方根 指数计算 对数计算等等

mean, sum,pow,sqrt,exp,log 等

def caculate_function():
    # # 设置PyTorch的打印选项，这里禁用了科学计数法（sci_mode=False），以便在打印张量时以常规格式显示数字。
    # torch.set_printoptions(sci_mode=False)
    # 默认是torch.int64，转换为float或double类型，否则运算函数会报错
    data = torch.randint(0, 10, (2, 3), dtype=torch.float32)
    print("data1-->,data1", data, data.dtype)
    # 1.均值
    print(data.mean())
    print(data.mean(dim=0))  # 按列计算均值
    print(data.mean(dim=1))  # 按行计算均值
    # 2.求和
    print(data.sum()) # print(torch.sum(data))
    print(data.sum(dim=0))
    print(data.sum(dim=1))
    # 3.求平方
    print(data.pow(2)) # print(torch.pow(data,2))
    # 4.求平方根
    print(data.sqrt())
    # 5.求指数 每个元素进行e^x计算
    print(data.exp())
    # 6.求对数
    print(data.log()) #以e为底的对数
    print(data.log2()) #以2为底的对数
if __name__ == '__main__':
    caculate_function()