SplitTensor# class SplitTensor[source]# Bases: Module Wrapper around torch.split __init__(size_or_sizes, dim)[source]# Methods __init__(size_or_sizes, dim) forward(X) __init__(size_or_sizes, dim)[source]# static __new__(cls, *args: Any, **kwargs: Any) → Any# Parameters args (Any) – kwargs (Any) – Return type Any forward(X)[source]#