dgs.utils.state.collate_tvt_tensors¶
- dgs.utils.state.collate_tvt_tensors(batch: list[torchvision.tv_tensors.Image | torchvision.tv_tensors.Mask | torchvision.tv_tensors.Video], *_args, **_kwargs) torchvision.tv_tensors.TVTensor | torchvision.tv_tensors.Image | torchvision.tv_tensors.Mask | torchvision.tv_tensors.Video [source]¶
Collate a batch of tv_tensors into a batched version of it.