torch.chunk()函數

將tensor按dim方向分割成chunks個tensor塊，返回的是一個元組。 示例： ...