Tensor ): data = (data,) for d in data: if isinstance(d, torch.Tensor): tensors.append(d.to(dl_device)) else: tensors.append(d) registered_e_tensors ...
確定! 回上一頁