pytorch每日一学30(torch.chunk())将一个tensor分为指定数量的块

第30个方法

torch.chunk(input, chunks, dim=0) → List of Tensors

此方法是将给定的input分为指定数量的块，并返回划分完以后每个块的视图

接下来介绍参数：

• input：将要被划分的tensor
• chunks：将要被划分的块的数量，注意是数量，所以此参数的类型应该为整型。
• dim：将要划分tensor的指定的维度，默认为0。

接下来看用法：

还可以指定维度为1：

注意，当a指定维度上的数据数量不能整除chunks时，最后一个块的数量会更小：

如果chunks大于a指定维度上的数据数量，那么可以认为按照chunks=a指定维度数据数量来进行操作（例如下图中chunks可以按照等于3进行操作）