赞
踩
PyTorch 的官方介绍是一个拥有强力GPU加速的张量和动态构建网络的库,其主要构件是张量,所以我们可以把 PyTorch 当做 NumPy 来用,PyTorch 的很多操作好 NumPy 都是类似的,但是因为其能够在 GPU 上运行,所以有着比 NumPy 快很多倍的速度。
看一下两者之间的转换:
- import torch
- import numpy as np
-
- # 创建一个 numpy ndarray
- numpy_tensor = np.random.randn(10, 20)
-
- #我们可以使用下面两种方式将numpy的ndarray转换到tensor上
- pytorch_tensor1 = torch.Tensor(numpy_tensor)
- pytorch_tensor2 = torch.from_numpy(numpy_tensor)
-
-
-
- #同时我们也可以使用下面的方法将 pytorch tensor 转换为 numpy ndarray
- # 如果 pytorch tensor 在 cpu 上
- numpy_array = pytorch_tensor1.numpy()
-
- # 如果 pytorch tensor 在 gpu 上
- numpy_array = pytorch_tensor1.cpu().numpy()
-
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。