以包含批量维度的图像张量为例,设张量 A 保存了 4 张,长和宽为 32 的三通道像素矩阵,则张量 A 的形状为 [4, 3, 32, 32](PyTorch将通道维度放在前面,即 (batch_size...同样的方式,张量 B 保存了另外 5 张,长和宽为 32 的三通道像素矩阵,张量 B 的形状为 [5, 3, 32, 32]。...(4, 3, 32, 32)
# 模拟图像张量B
b = torch.randn(5, 3, 32, 32)
# 在批量维度上合并张量A和B
cat_ab = torch.cat([a, b], dim...参数 tensors 中所有需要合并的张量必须是相同的数据类型
import torch
# 模拟图像张量A
a = torch.randint(0, 255, (4, 3, 32, 32))
# 模拟图像张量...element 1.
'''
非合并维度的长度必须一致
「从语法上来说,拼接合并操作可以在任意的维度上进行,唯一的约束是非合并维度的长度必须一致。」