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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
爱摸鱼的打工仔1 分钟前
【VLLM启动大模型共享内存不足-AI知识点】
人工智能
初心未改HD1 分钟前
深度学习之正则化技术详解
人工智能·深度学习
user29876982706541 分钟前
三、Skills 进阶:Fork 模式与上下文控制
人工智能
闵孚龙2 分钟前
Claude Code CLAUDE.md 用户指令覆盖层全解析:AI Agent 记忆系统、上下文工程、规则分层、团队协作与安全治理
人工智能·安全
X54先生(人文科技)2 分钟前
《元创力》纪实录·卷宗2.1 观测续篇试纸:当“社会性死亡”的叙事进入审核队列
人工智能·开源·ai写作·零知识证明
Hector_zh2 分钟前
逐浪 · 第七篇:Trae-SOLO 多端协同 —— 从安装到完成任务的完整流程
人工智能·trae
189228048616 分钟前
NV301固态MT29F32T08GWLBHD6-QJES:B
大数据·服务器·人工智能·科技·缓存
木雷坞8 分钟前
模型评测 Job 卡住:从 PodGroup 到镜像预检的排查记录
人工智能
大模型推理9 分钟前
Nano-vLLM 源码解读 - 8. Chunked Prefill
人工智能
zh252610 分钟前
深入 OpenViking:字节开源的 Agent 上下文数据库,解决了5 个问题
人工智能·开源