我遇到这么一个问题,就是说我要把一个列表的参数经过变换传入到一个函数的参数中,我以前从来没有这么写过,下面是一个基于pytorch的例子
import numpy as np
import torch
data=np.array([[0,0],[1,1],[2,2],[3,3],[4,4],[5,5],[6,6],[7,7],[8,8],[9,9]])
data_ls=[data,data,data]
dataset=torch.utils.data.TensorDataset(*[torch.FloatTensor(i) for i in data_ls])
可以看到我构造DataSet时,是不管data_ls中有多少个array的,这个刚好达到了我的要求