import platform import torch def to_shared_memory(tensors: tuple[torch.Tensor]): return [tensor.cpu() for tensor in tensors if tensor is not None] """ if platform.system() == "Windows": return [tensor.cpu() for tensor in tensors if tensor is not None] return [tensor.share_memory_() for tensor in tensors if tensor is not None] """ def to_device(tensors: tuple[torch.Tensor], device: torch.device): return [tensor.to(device) for tensor in tensors if tensor is not None]