SplitWithSizes ============== Splits the tensor into chunks chunks with specified sizes. **Abstract Signature:** ``SplitWithSizes(tensor: Tensor, split_sizes: List[int], dim: int = 0)`` .. raw:: html
torch.split_with_sizes
—
tf.split