本文简要介绍python语言中 torch.max
的用法。
用法:
torch.max(input) → Tensor
input(Tensor) -输入张量。
out(tuple,可选的) -两个输出张量的结果元组(最大值,max_indices)
返回
input
张量中所有元素的最大值。警告
与
max(dim=0)
不同,此函数产生确定性(子)梯度例子:
>>> a = torch.randn(1, 3) >>> a tensor([[ 0.6763, 0.7445, -2.2369]]) >>> torch.max(a) tensor(0.7445)
torch.max(input, dim, keepdim=False, *, out=None)
返回一个命名元组
(values, indices)
其中values
是给定维度dim
中input
张量的每一行的最大值。而indices
是找到的每个最大值的索引位置(argmax)。如果
keepdim
是True
,则输出张量的大小与input
相同,但在维度dim
中它们的大小为 1。否则,dim
被压缩(参见torch.squeeze()
),结果在比input
少 1 个维度的输出张量中。注意
如果缩减行中有多个最大值,则返回第一个最大值的索引。
例子:
>>> a = torch.randn(4, 4) >>> a tensor([[-1.2360, -0.2942, -0.1222, 0.8475], [ 1.1949, -1.1127, -2.2379, -0.6702], [ 1.5717, -0.9207, 0.1297, -1.8768], [-0.6172, 1.0036, -0.6060, -0.2432]]) >>> torch.max(a, 1) torch.return_types.max(values=tensor([0.8475, 1.1949, 1.5717, 1.0036]), indices=tensor([3, 0, 0, 1]))
torch.max(input, other, *, out=None) → Tensor
见
torch.maximum()
。
参数:
参数:
关键字参数:
相关用法
- Python PyTorch maximum用法及代码示例
- Python PyTorch matrix_rank用法及代码示例
- Python PyTorch matrix_exp用法及代码示例
- Python PyTorch matmul用法及代码示例
- Python PyTorch matrix_power用法及代码示例
- Python PyTorch masked_select用法及代码示例
- Python PyTorch maskrcnn_resnet50_fpn用法及代码示例
- Python PyTorch matrix_norm用法及代码示例
- Python PyTorch make_tensor用法及代码示例
- Python PyTorch monitored_barrier用法及代码示例
- Python PyTorch mean用法及代码示例
- Python PyTorch multinomial用法及代码示例
- Python PyTorch meshgrid用法及代码示例
- Python PyTorch mm用法及代码示例
- Python PyTorch mv用法及代码示例
- Python PyTorch min用法及代码示例
- Python PyTorch msort用法及代码示例
- Python PyTorch mode用法及代码示例
- Python PyTorch movedim用法及代码示例
- Python PyTorch minimum用法及代码示例
- Python PyTorch multi_dot用法及代码示例
- Python PyTorch mul用法及代码示例
- Python PyTorch movielens_25m用法及代码示例
- Python PyTorch multigammaln用法及代码示例
- Python PyTorch movielens_20m用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.max。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。