modulation.torch_utilities.collation.collate_pad_wrapped

modulation.torch_utilities.collation.collate_pad_wrapped(device: Optional[Union[device, str]] = device(type='cuda')) → callable

Parameters

device –

Returns

Return type