torch.dsplit — PyTorch 1.9.1 documentation
torch.dsplit(input, indices_or_sections) → List of Tensors. Splits input, a tensor with three or more dimensions, into multiple tensors depthwise according to indices_or_sections. Each split is a view of input. This is equivalent to calling torch.tensor_split (input, indices_or_sections, dim=2) (the split dimension is 1), except that if ...
DA: 52 PA: 47 MOZ Rank: 53