jactorch.data.collate.collate_v1#

Function for collate data generated from multiple workers. It supports collating data with variant length.

Classes

VarLengthCollateV1

Class VarLengthCollateV1

class VarLengthCollateV1[source]#

Bases: object

__call__(batch, process=False)[source]#

Call self as a function.

__init__(fields, mode='pad')[source]#

VarLengthCollate is deprecated in favor of VarLengthCollateV2.

Parameters:

  • fields – the fields to be collated with variant length.

  • mode – the collate mode.

__new__(**kwargs)#
On this page