torch.chunk. torch. chunk (input, chunks, dim=0) → List of Tensors. Attempts to split a tensor into the specified number of chunks. Each chunk is a view of ...
確定! 回上一頁