jactorch.data.collate.utils

Classes

VarLengthCollateMode

Functions

use_shared_memory()
user_scattered_collate(batch)	A helper function indicating that no collation needs to be done.

Class VarLengthCollateMode

class VarLengthCollateMode

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 from_string(value)

Parameters:

value (str | JacEnum)

Return type:

JacEnum

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

use_shared_memory()

user_scattered_collate(batch)

A helper function indicating that no collation needs to be done.