torch.Tensor.chunk

Tensor.chunk(chunks, dim=0) Tensor 列表

参见 torch.chunk()

本页目录