SplitTensor#

class SplitTensor[source]#

Bases: Module

Wrapper around torch.split

__init__(size_or_sizes, dim)[source]#

Methods

__init__(size_or_sizes, dim)

forward(X)

__contains__(key: str) bool#
Parameters

key (str) –

Return type

bool

__getitem__(key: Any) _MockObject#
Parameters

key (Any) –

Return type

_MockObject

__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]#