torch的maximum与max以及导出onnx

  • 转onnx的时候maximum不支持,换成max,效果是一样的

获得 Tensor 元素中的最大值、最小值

1、torch.max()

torch.max(input, dim, keepdim=False)
返回命名元组(最大值,最大值索引),最大值是给定维度中的最大值,索引为在对应维度中的索引
当有多个最大值时,返回第一个最大值的索引
当没有指定维度 dim 时,则返回所有元素中的最大值(single value)
input:输入张量
dim:指定的维度
keepdim:默认 False,输出是否保留原始维度
在这里插入图片描述
torch 中还给出了直接获取最大值索引的方法 torch.argmax()

其用法和 torch.max() 一致,且结果即为 torch.max() 返回的第二个元组(索引的元组)

2、torch.min()

torch.min(input, dim, keepdim=False)

返回命名元组(最小值,最小值索引),最小值是给定维度中的最小值,索引为在对应维度中的索引

当有多个最小值时,返回第一个最小值的索引

当没有指定维度 dim 时,则返回所有元素中的最小值(single value)

input:输入张量

dim:指定的维度

在这里插入图片描述

torch 中还给出了直接获取最大值索引的方法 torch.argmin()

其用法和 torch.min() 一致,且结果即为 torch.min() 返回的第二个元组(索引的元组)

3、 torch.maximum()

计算输入张量成对比较的最大值

如果比较元素包含 NaN,则返回 NaN

torch.maximum(input,other)

input:输入张量

other:第二个输入张量,应与 input 同形状

在这里插入图片描述

4、torch.minimum()

计算输入张量成对比较的最小值

如果比较元素包含 NaN,则返回 NaN

torch.minimum(input,other)

input:输入张量

other:第二个输入张量,应与 input 同形状

在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值