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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
lpfasd1233 小时前
2026年第30周科技社区趋势周报:开放权重之争与AI Agent的破局
人工智能·科技
佛光芳林5 小时前
AI 情报局:用 PowerMem + SeekDB 做一个多 Agent 记忆小游戏
人工智能
通问AI5 小时前
2026年AI短剧技术现状:全链路AIGC生产已达95%,但用户完播率不足15%
人工智能·aigc
_Jimmy_6 小时前
Agent引用数据库知识过时的增量同步方案
人工智能·python·langchain
qq_454245036 小时前
Systemprompt 体系全览:形式化公理驱动的分层系统设计
人工智能
大龄码农有梦想6 小时前
传统的 BPMN 工作流审批和 AI 工作流有什么区别?
人工智能·流程引擎·工作流·ai agent·ai工作流·审批流·智能体平台
DO_Community6 小时前
Claude Opus 5 现已上线 DigitalOcean AI 推理云
人工智能·llm·agent·claude
幸福指北6 小时前
🚀 开源了,一个人 + AI 肝出一个 AI 终端 | AShell 技术分享
运维·人工智能·ai·终端
可以飞的话6 小时前
一、机器学习概述
人工智能·机器学习
sunneo6 小时前
磐石2.0发布,科学建模新突破
人工智能