jactorch.data.layout#
Classes
Functions
Class DataLayout
- class DataLayout[source]#
Bases:
object
- __new__(**kwargs)#
- decl(key, spec)[source]#
- Parameters:
key (str)
spec (str | DataLayoutSpec)
Class DataLayoutSpec
Class DataLayoutType
- class DataLayoutType[source]#
Bases:
JacEnum
- __new__(value)#
- classmethod assert_valid(value)#
Assert if the value is a valid choice.
- classmethod choice_names()#
Returns the list of the name of all possible choices.
- classmethod choice_objs()#
Returns the list of the object of all possible choices.
- classmethod choice_values()#
Returns the list of the value of all possible choices.
- classmethod is_valid(value)#
Check if the value is a valid choice.
- classmethod type_name()#
Return the type name of the enum.
- CONCAT = 'concat'#
- PAD = 'pad'#
- PAD2D = 'pad2d'#
- PADIMAGE = 'padimage'#
- SKIP = 'skip'#
- STACK = 'stack'#
Functions