[ 输入参数用法不一致 ]torch.utils.data.TensorDataset¶

torch.utils.data.TensorDataset ¶

          torch.utils.data.TensorDataset(*tensors)

         

          paddle.io.TensorDataset(tensors)

         

paddle 参数和 torch 参数用法不一致，具体如下：

PyTorch	PaddlePaddle	备注
*tensors	tensors	输入的 Tensor， PyTorch 是可变参数用法， Paddle 是列表或元组，需要转写

           # PyTorch 写法
torch.utils.data.TensorDataset(tensor1, tensor2, tensor3)

# Paddle 写法
paddle.io.TensorDataset([tensor1, tensor2, tensor3])