这段代码展示了如何使用numpy创建一个包含多个随机3x224x224尺寸矩阵的列表，然后将这个列表转换为numpy数组，并进一步转化为PyTorch张量。时间戳t1用于记录转换过程的时间。
input = []
for i in range(10):
x = np.random.rand(3, 224, 224)
input.append(x)
t1 = time.time()
input = np.asarray(input)
input = torch.from_numpy(input)
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
