The size of tensor a (4) must match the size of tensor b (2) at non-singleton dimension 0

最新推荐文章于 2024-12-23 10:06:03 发布

AI算法网奇

最新推荐文章于 2024-12-23 10:06:03 发布

阅读量4.5w

点赞数 2

CC 4.0 BY-SA版权

分类专栏： pytorch知识宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/82714120

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

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

当尝试将尺寸为4的张量a与尺寸为2的张量b在维度0上操作时，会出现尺寸不匹配的错误。这个问题通常发生在深度学习运算中，需要确保张量在指定维度上有相同的尺寸或使用恰当的形状转换。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

The size of tensor a (4) must match the size of tensor b (2) at non-singleton dimension 0

a1与a2维度不一样就会报这个错：

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, 1, steps=2).repeat(16,1).view(16,2)
b = torch.linspace(1, 256, steps=256).view(8,8, 4)

print(np.array(a))
print(b<5)

#取单个值，
a=[[0,0,0,1],[0,0,0,2]]#维度变了
a1=np.asarray([0,0,0,1])#维度变了
a2=np.asarray([0,2])#维度变了
b[a1,a2]=0

print(b)