import numpy as np
import torch
ls = []
ls = np.array(ls) # list -> nupmy
ls = np.expand_dims(ls, axis=1)
# numpy改维度可以查询np.expand_dims()函数的用法
ls = torch.from_numpy(ls) # numpy -> torch
list转numpy,numpy改维度再转torch
最新推荐文章于 2022-08-25 20:50:43 发布