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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
心中有国也有家9 分钟前
hccl 架构拆解:昇腾集合通信库到底在做什么?
人工智能·经验分享·笔记·分布式·算法·架构
這花開嗎19 分钟前
试了一圈配音网站,说说我的感受
人工智能·语音识别
w_t_y_y22 分钟前
AI应用demo(二)打造个人的code agent
人工智能·语音识别
Raink老师27 分钟前
【AI面试临阵磨枪-60】微服务下 AI 能力如何封装、网关、限流、监控
人工智能·微服务·面试
ApiHug34 分钟前
Mintlify、Stainless & ApiHug 在AI 时代的战略意义
人工智能
九皇叔叔36 分钟前
Spring-Ai-Alibaba [04] 04-llm-platform-custom-demo
java·人工智能·spring
ai生成式引擎优化技术36 分钟前
DLOS Kernel v0.5:从多Agent系统到AI操作系统内核的拐点
人工智能
CHEN5_0240 分钟前
深入理解 RAG(检索增强生成):核心流程、技术选型与进阶实战
人工智能·rag
@蔓蔓喜欢你41 分钟前
团队协作工具:提升开发效率的利器
人工智能·ai
2601_957879331 小时前
短视频矩阵的数据驱动运营:从流量监测到内容迭代的完整技术链路
大数据·矩阵·音视频