pytorch之---max()函数

形式： torch.max(input) → Tensor
返回输入tensor中所有元素的最大值：

a = torch.randn(1, 3)
>>0.4729 -0.2266 -0.2085
torch.max(a) #也可以写成a.max()
>>0.4729

形式： torch.max(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)
按维度dim 返回最大值，并且返回索引。
torch.max(a,0)返回每一列中最大值的那个元素，且返回索引（返回最大元素在这一列的行索引）。返回的最大值和索引各是一个tensor，一起构成元组(Tensor, LongTensor)

a = torch.randn(3,3)
>>
0.2252 -0.0901  0.5663
-0.4694  0.8073  1.35960.1073 -0.7757 -0.8649torch.max(a,0)
>>
(0.22520.80731.3596
[torch.FloatTensor of size 3]
, 011
[torch.LongTensor of size 3]
)

torch.max(a,1)返回每一行中最大值的那个元素，且返回其索引（返回最大元素在这一行的列索引）

a = torch.randn(3,3)
>>
0.2252 -0.0901  0.5663
-0.4694  0.8073  1.35960.1073 -0.7757 -0.8649torch.max(a,1)
>>
(0.56631.35960.1073
[torch.FloatTensor of size 3]
, 220
[torch.LongTensor of size 3]
)

拓展：
torch.max()[0]，只返回最大值的每个数
troch.max()[1]，只返回最大值的每个索引
torch.max()[1].data 只返回variable中的数据部分（去掉Variable containing:）
torch.max()[1].data.numpy() 把数据转化成numpy ndarry
torch.max()[1].data.numpy().squeeze() 把数据条目中维度为1 的删除掉

转载:https://blog.csdn.net/Z_lbj/article/details/79766690

pytorch之---max()函数相关推荐

Pytorch中tensor维度和torch.max()函数中dim参数的理解
Pytorch中tensor维度和torch.max()函数中dim参数的理解维度参考了 https://blog.csdn.net/qq_41375609/article/details/106 ...
PyTorch max()函数取最大值
max() 函数本人在复现 LeNet5 网络进行 MNIST 数据集的训练,在模型预测阶段,需要用到 max() 函数进行结果的处理.将模型输出的 fc7 层的 10 个结果,变为最后的一个最大概 ...
如何使用PyTorch torch.max（）
In this article, we'll take a look at using the PyTorch torch.max() function. 在本文中,我们将介绍如何使用PyTorch ...
Pytorch：.max(0)和.max(1)的区别？
Pytorch 中.max(0)和.max(1)的区别是什么? 当我们有一个形状为 ( m , n ) (m, n) (m,n) 的 Tensor x x x 时,其中 m m m 表示行数, n n ...
pandas使用max函数和min函数计算dataframe日期（时间）数据列中最大日期和最小日期对应的数据行(maximum and minimum date or time row)
pandas使用max函数和min函数计算dataframe日期(时间)数据列中最大日期和最小日期对应的数据行(maximum and minimum date or time row in data ...
R语言max函数min函数计算各种数据对象最大值最小值实战
R语言max函数min函数计算各种数据对象最大值最小值实战目录 R语言max函数min函数计算各种数据对象最大值最小值实战 #基本语法
Python max() 函数
描述 max() 函数返回给定参数的最大值,参数可以为序列. 语法以下是 max() 函数的语法: max( x, y, z, .... ) 参数 x -- 数值表达式. y -- 数值表达式. z ...
MIN()与MAX()函数的注意事项
MIN()与MAX()函数的注意事项定义: MIN()函数用于返回一个列范围内的最小非空值:MAX()函数用于返回最大值.这两个函数可以用于大多数的数据类型,返回的值根据对不同数据类型的排序规 ...
数据库-聚合函数-max函数
avg函数 mysql> select avg(price) from shop; +‐‐‐‐‐‐‐‐‐‐‐‐+ | avg(price) | +‐‐‐‐‐‐‐‐‐‐‐‐+ | 6.110000 ...

pytorch之---max()函数

pytorch之---max()函数相关推荐

最新文章

热门文章