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)
【pytorch】截取图片 -> 合并为矩阵
最新推荐文章于 2025-01-13 17:12:07 发布
这段代码展示了如何使用numpy创建一个包含多个随机3x224x224尺寸矩阵的列表,然后将这个列表转换为numpy数组,并进一步转化为PyTorch张量。时间戳t1用于记录转换过程的时间。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
674

被折叠的 条评论
为什么被折叠?



