AI认知推理
定义:torch.chunk(tensor,chunk,dim)
将一个张量数据平均分割指定数据块
参数:
tensor : 张量数据
chunk:分块数量
dim:切分的维度,dim = 0:按行分割,行数会变; dim = 1:按列分割,列数会变
定义:torch.cat(tensors,dim)
将多个张量数据进行拼接
tensors:多个张量数据,已数组的方式传入
dim:拼接的维度,dim = 0:按行拼接,张量的行数相等; dim = 1:按列拼接,张量的列数相等