Pytorch学习之：索引出一个 tensor 中的最大值及其所在的位置 torch.argmax

代码torch.argmax 可以返回一个值在张量中展平后的索引位置在这种条件下要得到最大的值，要先把张量展平才能得到相应的值t1 = torch.Tensor([[1,2,3,4],[2,2,3,4]])t1tensor([[1., 2., 3., 4.],[2., 2., 3., 4.]])index = torch.argmax(t1)indextensor(3)t1.view(-1)[in

暖仔会飞

15990人浏览 · 2022-01-23 20:33:42

暖仔会飞  ·  2022-01-23 20:33:42 发布

代码

• torch.argmax 可以返回一个值在张量中展平后的索引位置
• 在这种条件下要得到最大的值，要先把张量展平才能得到相应的值

t1 = torch.Tensor([[1,2,3,4],[2,2,3,4]])
t1

tensor([[1., 2., 3., 4.],
    [2., 2., 3., 4.]])

index = torch.argmax(t1)
index

tensor(3)

t1.view(-1)[index]

tensor(4.)

参考文献

pytorch查找矩阵中最大元素的值和索引

# pytorch # 深度学习 # python

华为云开发者联盟

为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。

加入社区

更多推荐

• · openEuler系统通过shell脚本安装openGauss 5.0.0企业版
• · 一文了解npm install -g和npm install --save-dev的关系
• · 【FAQ】HarmonyOS SDK 闭源开放能力 —Map Kit（2）

openEuler系统通过shell脚本安装openGauss 5.0.0企业版

华为云开发者联盟

一文了解npm install -g和npm install --save-dev的关系

华为云开发者联盟

【FAQ】HarmonyOS SDK 闭源开放能力 —Map Kit（2）

华为云开发者联盟

欢迎加入社区

取消 确定