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.