以包含批量维度的图像张量为例,设张量 A 保存了 4 张,长和宽为 32 的三通道像素矩阵,则张量 A 的形状为 [4, 3, 32, 32](PyTorch将通道维度放在前面,即 (batch_size...同样的方式,张量 B 保存了另外 5 张,长和宽为 32 的三通道像素矩阵,张量 B 的形状为 [5, 3, 32, 32]。...>
cat_ab = torch.cat([a, b], dim = 0)
RuntimeError: Expected object of scalar type long int but got...比如图像张量形状为 [4, 3, 32, 32] 和形状为 [5, 1, 32, 32] 的张量不能直接在批量维度上进行合并,因为两个图像张量的通道维度的长度不一致,一个是单通道(channels =...Got 1 and 3 in dimension 1
'''
References:
1. 《TensorFlow深度学习》