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

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

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

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

而矩阵的单个数值可赋值:

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
模域ModelSphere25 分钟前
长回复吐到一半断了,流式推理还能接着写吗?
人工智能
用户19172912708328 分钟前
Agent 运行时还是预览版,企业怎么上生产?准入 4 查与隔离回滚实战
人工智能
海螺_AIHCC31 分钟前
开源模型挤着发了一圈,先别比榜单,算算显存
人工智能
米小虾31 分钟前
给正在干活的 Agent 插一句话:steering 的原理与实现
人工智能·agent
柯南466842 分钟前
【AI工程师精讲】32:Text2SQL:让 AI 查数据库,难点八成不在生成 SQL 上
人工智能·ai编程
丘山录1 小时前
本地部署 PLaMo,llama.cpp 完整教程,不用显卡
人工智能
播播资源1 小时前
同账户实测 11 个多模态模型:图像 0.03 元起、竖屏视频 8 元一条,最贵和最便宜差 267 倍
大数据·人工智能·深度学习
Linye1 小时前
AI 越跑越偏?从 500 行臃肿规则到 80 行核心约束:多 Agent 项目中的「规则瘦身与防腐」四步排查法
人工智能·架构
Eyfcom1 小时前
〖AI物流落地 06〗AI Copilot 在 TMS 调度系统中的应用:Text-to-SQL、意图拆解与自动调参
人工智能·物流系统·快递系统·跨境物流快递系统·ai物流落地
域限奇点访客1 小时前
怎么让AI稳定输出可解析的问卷结构?schema约束、失败重试与降级策略
人工智能