index 0 does not match the shape of the indexed tensor [8, 8, 4] at index 0

最新推荐文章于 2024-07-13 07:15:00 发布

AI算法网奇

最新推荐文章于 2024-07-13 07:15:00 发布

阅读量6.4k

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/82713000

版权

pytorch知识宝典专栏收录该内容

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客介绍了在遇到`index 0 does not match the shape of the indexed tensor [8, 8, 4] at index 0`这种错误时，如何进行问题定位和解决。关键点在于，解决方案涉及到将b的索引从nparray转换为tensor。" 12916343,1932395,AIX系统fsck修复：从故障到解析,"['AIX操作系统', '文件系统构架', '故障排查', 'fsck命令']

摘要由CSDN通过智能技术生成

index 0 does not match the shape of the indexed tensor [8, 8, 4] at index 0

解决后的；

b的索引是nparray，不是tensor，

b[np.asarray(a)]

import torch
import numpy as np

# 80 10 5 3:4
# 80 3 11 11
import copy
a = torch.linspace(-4, -1, steps=4)



a = torch.linspace(0, 7, steps=8).repeat(4,1).view(2,16).type(torch.uint8)
b = torch.linspace(1, 256, steps=256).view(8,8, 4)


#取单个值，
# a=[[0,0,0,1],[0,0,0,2]]#维度变了

b[np.asarray(a)]=0

print(b)
# if b[np.asarray(a)]==b[np.asarray(c)]:
#     print("11111111111")
# print(np.array(a))
#
# b[np.array(a)]=0

解决之前的：