51工具盒子

依楼听风雨
笑看云卷云舒,淡观潮起潮落

PyTorch Tensor 运算函数

PyTorch 为每个张量封装很多实用的计算函数,例如计算均值、平方根、求和等等。

  1. mean 函数用于计算张量均值
  2. sum 函数用于计算张量和
  3. pow 函数用于计算张量平方
  4. sqrt 函数用于计算张量平方根
  5. exp 函数用于计算张量e的多少次方
  6. log 函数用于以e为底求对数

示例代码:

import torch


def test():

    data = torch.randint(0, 10, [2, 3], dtype=torch.float64)
    print(data)
    print('-' * 50)

    # 1. 计算均值
    # 注意: tensor 必须为 Float 或者 Double 类型
    print(data.mean())
    print(data.mean(dim=0))  # 按列计算均值
    print(data.mean(dim=1))  # 按行计算均值
    print('-' * 50)

    # 2. 计算总和
    print(data.sum())
    print(data.sum(dim=0))
    print(data.sum(dim=1))
    print('-' * 50)

    # 3. 计算平方
    print(data.pow(2))
    print('-' * 50)

    # 4. 计算平方根
    print(data.sqrt())
    print('-' * 50)

    # 5. 指数计算, e^n 次方
    print(data.exp())
    print('-' * 50)

    # 6. 对数计算
    print(data.log())  # 以 e 为底
    print(data.log2())
    print(data.log10())


if __name__ == '__main__':
    test()
赞(0)
未经允许不得转载:工具盒子 » PyTorch Tensor 运算函数