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

torch.utils.data.TensorDataset

torch.utils.data.TensorDataset(*tensors)

paddle.io.TensorDataset

paddle.io.TensorDataset(tensors)

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

参数映射

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

转写示例

tensors：输入的 Tensor

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

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