modulation.torch_utilities.collation.collate_pad_wrapped

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

device

Returns

Return type