Pytorch学习：torch.max(input，dim，keepdim=False)-Toy模板网

这篇具有很好参考价值的文章主要介绍了Pytorch学习：torch.max(input，dim，keepdim=False)。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

torch.max()

torch.max(input) → Tensor：返回 input 张量中所有元素的最大值。
注意输入的必须是张量形式，输出的也为张量形式
Pytorch学习：torch.max(input，dim，keepdim=False),pytorch学习,pytorch,学习,人工智能

当输入为tuple类型时，会报错，需要将输入改为tensor类型，输出也为tensor类型

torch.max()：官方文档
torch.max（input，dim，keepdim=False，*，out=None）
主要参数：

input（Tensor）-输入张量。

dim（int）-要减少的维度。

keepdim（bool）-输出张量是否保留了 dim 。默认值： False 。
关键字参数：

out（tuple，optional）-两个输出张量的结果元组（max，max_indices）

dim

对于二维数组来说，dim=0为行，dim=1为列
在torch.max()中代表要减少的维度（dimension）

import torch

a = torch.tensor([1, 2, 3, 4])
max = torch.max(a, dim=0)
print(max)

对于以上程序，由于只存在行，所以torch.max(a, dim=0)只能减少的维度为行向量，即dim=0
Pytorch学习：torch.max(input，dim，keepdim=False),pytorch学习,pytorch,学习,人工智能
如果 max = max = torch.max(a, dim=1)，则会报错：维度错误

注：如果在减少的行中存在多个最大值，则返回第一个最大值的索引。

import torch

a = torch.tensor([4, 2, 3, 4])
max = torch.max(a, dim=0)
print(max)

Pytorch学习：torch.max(input，dim，keepdim=False),pytorch学习,pytorch,学习,人工智能

keepdim

输出张量是否保留了 dim，即设置是否保留torch.max(input, dim=0, keepdim=True) 中需要消去的dim。

如果 keepdim 是 True ，则输出张量的大小与 input 相同，除了在维度 dim 中它们的大小为1。

dim=0

二维数组中dim=0代表行，torch.max(a, dim=0)代表消去行，求每列的最大值，keepdim=True则代表保留行

import torch

a = torch.tensor([[1, 2, 3, 4],
                  [4, 1, 2, 3],
                  [6, 2, 3, 4],
                  [3, 4, 5, 9]])

# dim = 0
max1_1 = torch.max(a, dim=0, keepdim=False)
max1_2 = torch.max(a, dim=0, keepdim=True)
print(max1_1)
print(max1_2)

Pytorch学习：torch.max(input，dim，keepdim=False),pytorch学习,pytorch,学习,人工智能
dim=0，消去的维数为行，即求每列的最大值
keepdim=False，vlaues=tensor([6, 4, 5, 9])有一个中括号
keepdim=True，vlaues=tensor([[6, 4, 5, 9]])有两个中括号

indices代表最大值所处的位置（第一列第三个：2，第一列第四个：3，第三列第四个：3，第四列第四个：3）

dim=1

二维数组中dim=1代表列，torch.max(a, dim=0)代表消去列，求每行的最大值，keepdim=True则代表保留列

import torch

a = torch.tensor([[1, 2, 3, 4],
                  [4, 1, 2, 3],
                  [6, 2, 3, 4],
                  [3, 4, 5, 9]])

# dim = 1
max2_1 = torch.max(a, dim=1, keepdim=False)
max2_2 = torch.max(a, dim=1, keepdim=True)
print(max2_1)
print(max2_2)

Pytorch学习：torch.max(input，dim，keepdim=False),pytorch学习,pytorch,学习,人工智能

dim=1，消去的维数为列，即求每行的最大值
keepdim=False，vlaues=tensor([4, 4, 6, 9])有一个中括号
keepdim=True，vlaues=tensor([[4], [4], [6], [9]])有两个中括号