custom_collate

lightautoml.text.utils.custom_collate(batch)[source]

Puts each data field into a tensor with outer dimension batch size.

Return type:

Tensor