python将列表元素解释为函数参数

我遇到这么一个问题,就是说我要把一个列表的参数经过变换传入到一个函数的参数中,我以前从来没有这么写过,下面是一个基于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的,这个刚好达到了我的要求

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值