PyTorch——各种常用函数总结

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

---

torch.max

torch.min与torch.max完全类似

单参数

• 用法

  torch.max(input) -> Tensor

  • input: 一个Tensor的对象
  • return: 返回input变量中的最大值

多参数

• 用法

  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: 指明是否