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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
DO_Community2 分钟前
DigitalOcean携手Persistent达成战略合作,让 AI 更亲民、更易扩展
大数据·人工智能·ai·llm·区块链
70asunflower4 分钟前
torch.manual_seed()介绍
人工智能·pytorch·python
塔能物联运维10 分钟前
年底回顾数据支撑:塔能科技助力钢结构安全管理优化
人工智能
自己的九又四分之三站台13 分钟前
微软AI库Microsoft.Extensions.AI的发展历史和背景介绍
人工智能·microsoft
程序员泠零澪回家种桔子14 分钟前
MCP协议(Model Context Protocol)及其在AI大模型系统中的作用
人工智能·ai
wfeqhfxz258878219 分钟前
柿子与桃子目标检测识别-YOLO11-seg-HGNetV2改进实现
人工智能·目标检测·计算机视觉
ZCXZ12385296a26 分钟前
基于YOLOv10n-LSDECD的多类别交通目标检测系统_行人_自行车及交通信号灯识别
人工智能·yolo·目标检测
AI科技星36 分钟前
统一场论理论下理解物体在不同运动状态的本质
人工智能·线性代数·算法·机器学习·概率论
乾元40 分钟前
数据为王——安全数据集的清洗与特征工程
大数据·网络·人工智能·安全·web安全·机器学习·架构