pytorch的二次索引矩阵无法赋值问题

最近在研究中发现torch一个问题,即torch的二次索引的矩阵无法赋值。

具体来说,给定相同的初始常数矩阵aiou_target 矩阵, 以及另一iou矩阵,直接赋值是没问题的。

然而,当对iou_target矩阵进行二次索引时,即idx矩阵和tp矩阵,可以看到赋值并未起到作用:

而矩阵的单个数值可赋值

根据以上观测,在二次索引赋值时,需通过中间变量tp进行。上述过程可以实现为:

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
仙女修炼史1 天前
CNN的捷径学习Shortcut Learning in Deep Neural Networks
人工智能·学习·cnn
企业架构师老王1 天前
数据安全优先:企业级智能体私有化部署完整方案与最佳实践
人工智能·ai
北漂Zachary1 天前
AI训练硬件选型:GPU算力梯队全解析
人工智能·gpu算力
CS创新实验室1 天前
CS实验室行业报告:机器人领域就业分析报告
大数据·人工智能·机器人
才兄说1 天前
机器人二次开发机器狗巡检?绕行率低于10%
人工智能
QYR-分析1 天前
机器人物体检测/追踪传感器行业解析及市场机遇
人工智能·机器人
Pushkin.1 天前
数仓 AI Coding 应用实战:基于 AI 的数仓效能提升
人工智能·数仓ai提效
罗不俷1 天前
【机器学习】(一)机器学习入门概念
人工智能·机器学习
lizz6661 天前
Hermes-Agent:配置gateway网关,chat交互入口(钉钉Dingtalk)
人工智能