AI认知推理

pytorch中torch.chunk和torch.cat方法
贵鬼2025-01-31发布
98 0

定义:torch.chunk(tensor,chunk,dim)

将一个张量数据平均分割指定数据块

参数:

tensor :  张量数据

chunk:分块数量

dim:切分的维度,dim = 0:按行分割,行数会变; dim = 1:按列分割,列数会变


定义:torch.cat(tensors,dim)

将多个张量数据进行拼接

参数:

tensors:多个张量数据,已数组的方式传入

dim:拼接的维度,dim = 0:按行拼接,张量的行数相等; dim = 1:按列拼接,张量的列数相等