PyTorch_张量运算函数

import torch 
import numpy as np 

# 均值
def test01():
    torch.manual_seed(0)
    data = torch.randint(0, 10, [2, 3], dtype=torch.float64)
    
    # 默认对所有的数据计算均值
    print(data.mean())

    # 按指定的维度计算均值
    print(data.mean(dim = 0))
    print(data.mean(dim = 1))

# 求和
def test02():
    torch.manual_seed(0)
    data = torch.randint(0, 10, [2, 3], dtype=torch.float64)

    print(data.sum())

    print(data.sum(dim = 0))  # 列
    print(data.sum(dim = 1))  # 行

# 平方
def test03():
    torch.manual_seed(0)
    data = torch.randint(0, 10, [2, 3], dtype=torch.float64)

    data = data.pow(2)
    print(data)

# 平方根
def test04():
    torch.manual_seed(0)
    data = torch.randint(0, 10, [2, 3], dtype=torch.float64)
    data = data.sqrt()
    print(data)

# e多少次平方
def test05():
    torch.manual_seed(0)
    data = torch.randint(0, 10, [2, 3], dtype=torch.float64)
    data = data.exp()
    print(data)

# 对数
def test06():
    torch.manual_seed(0)
    data = torch.randint(0, 10, [2, 3], dtype=torch.float64)

    data = data.log()  # 默认以e为底
    print(data)

    data = data.log2()  # 以2为底
    print(data)

    data = data.log10()  # 以10为底
    print(data)

if __name__ == "__main__":
    test06()