[a9z3dzqs32.png]
合并
合并是指将多个张量在某个维度上合并成一个张量。...拼接
在 PyTorch 中,可以通过 torch.cat(tensors, dim = 0) 函数拼接张量,其中参数 tensor 保存了所有需要合并张量的序列(任何Python的序列对象,比如列表、...以包含批量维度的图像张量为例,设张量 A 保存了 4 张,长和宽为 32 的三通道像素矩阵,则张量 A 的形状为 [4, 3, 32, 32](PyTorch将通道维度放在前面,即 (batch_size...a, b], dim = 0)
RuntimeError: Expected object of scalar type long int but got scalar type float for sequence.../repeat_function.py", line 9, in
cat_ab = torch.cat([a, b], dim = 0)
RuntimeError: Sizes