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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
Rorsion17 小时前
对优化器的改进
人工智能·机器学习
2601_9492210317 小时前
边缘智算加速重构算力格局,微模块技术筑牢低延时基础设施底座
大数据·人工智能·重构
木易 士心17 小时前
AI辅助开发:前端“加速器”还是后端“稳定器”?——基于项目类型与用户规模的实战指南
人工智能·后端
欧阳子遥17 小时前
OpenCV 复杂背景下的轮廓提取
人工智能·opencv·计算机视觉
老鱼说AI17 小时前
CUDA架构与高性能程序设计:多维网格与数据
c++·人工智能·深度学习·神经网络·机器学习·语言模型·cuda
badhope17 小时前
GitHub超有用项目推荐:skill仓库--用技能树打造AI超频引擎
java·开发语言·前端·人工智能·python·重构·github
网安情报局17 小时前
2026网络安全六大确定性趋势
大数据·人工智能·网络安全
syounger17 小时前
当S/4HANA不再是唯一答案:CIO如何重新定义ERP转型路径
人工智能
yhdata17 小时前
聚焦半导体关键部件:磁悬浮无轴泵市场前景明朗,2032年规模逼近15.12亿元
大数据·人工智能