SplitCopy

Splits the tensor into chunks (Copy).

Abstract Signature:

SplitCopy(tensor: Tensor, split_size_or_sections: Union[int, List[int]], dim: int = 0)

PyTorchJAX (Core)

PyTorch

API: torch.split_copy

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (torch_split_converter)