torch.chunk(input, chunks, dim=0) → List of Tensors
功能:将数组拆分为特定数量的块
输入:
input
:待拆分的数组chunks
:拆分的块数,指定为几,就拆成几dim
:拆分的维度,默认沿第1维度拆分
torch.chunk(input, chunks, dim=0) → List of Tensors
功能:将数组拆分为特定数量的块
输入:
input
:待拆分的数组chunks
:拆分的块数,指定为几,就拆成几dim
:拆分的维度,默认沿第1维度拆分