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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
骚戴4 分钟前
2025 Python AI 实战:零基础调用 LLM API 开发指南
人工智能·python·大模型·llm·api·ai gateway
Cherry的跨界思维6 分钟前
【AI测试全栈:质量模型】4、新AI测试金字塔:从单元到社会的四层测试策略落地指南
人工智能·单元测试·集成测试·ai测试·全栈ai·全栈ai测试·社会测试
亚马逊云开发者24 分钟前
使用Amazon Nova模型实现自动化视频高光剪辑
人工智能
Tony Bai30 分钟前
Go 的 AI 时代宣言:我们如何用“老”原则,解决“新”问题?
开发语言·人工智能·后端·golang
卤代烃42 分钟前
🦾 可为与不可为:CDP 视角下的 Browser 控制边界
前端·人工智能·浏览器
ggabb1 小时前
海南封关:锚定中国制造2025,破解产业转移生死局
大数据·人工智能
_XU1 小时前
AI工具如何重塑我的开发日常
前端·人工智能·深度学习
Blossom.1182 小时前
Prompt工程与思维链优化实战:从零构建动态Few-Shot与CoT推理引擎
人工智能·分布式·python·智能手机·django·prompt·边缘计算
zxsz_com_cn2 小时前
设备预测性维护典型案例:中讯烛龙赋能高端制造降本增效
人工智能
人工智能培训3 小时前
图神经网络初探(1)
人工智能·深度学习·知识图谱·群体智能·智能体