PyTorch——各种常用函数总结

Posted on

PyTorch封装了很多有用的函数,本文主要介绍介绍其中常用的函数

torch.max

torch.min与torch.max完全类似

单参数

  • 用法

    1
    torch.max(input) -> Tensor
    • input: 一个Tensor的对象
    • return: 返回input变量中的最大值

多参数

  • 用法

    1
    torch.max(input, dim, keepdim=False, out=None) -> tuple[Tensor, Tensor]
    • input: 一个Tensor的对象
    • dim: 指明维度
      • dim=0: 生成的结果是第一维的数据为1, 对每个元素, 当前数据是遍历第一维的数据后的最大值
        • 如果数据为2维, 则搜索每一列中最大的那个元素, 且返回最大元素的行索引(实际上相当于对每个列我们要求出来一个数,这个数是遍历第一维(行)得到的), 每列返回一个行索引(该索引就是当前列中数字最大的行)
        • input 为 (2,3), 则返回 (1,3)
      • dim=1:
        • 如果数据为2维, 则搜索每一行中最大的那个元素, 且返回最大元素的列索引(实际上相当于对每个行我们要求出来一个数,这个数是遍历第2维(列)得到的), 每列返回一个列索引(该索引就是当前行中数字最大的列)
        • input 为 (2,3), 则返回 (2,1)
    • keepdim: 指明是否