Pyorch之numpy与torch之间相互转换方式
numpy中的ndarray转化成pytorch中的tensor : torch.from_numpy()
pytorch中的tensor转化成numpy中的ndarray : numpy()
代码
import numpy as np
import torch
np_arr = np.array([1,2,3,4])
tor_arr=torch.from_numpy(np_arr)
tor2numpy=tor_arr.numpy()
print('\nnumpy\n',np_arr,'\ntorch\n',tor_arr,'\nnumpy\n',tor2numpy)
输出
numpy
[1 2 3 4]
torch
tensor([1, 2, 3, 4], dtype=torch.int32)
numpy
[1 2 3 4]
以上这篇Pyorch之numpy与torch之间相互转换方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持软件开发网。
您可能感兴趣的文章:python、PyTorch图像读取与numpy转换实例Pytorch中的variable, tensor与numpy相互转化的方法pytorch模型预测结果与ndarray互转方式
相关文章
Vanora
2020-10-01
Adalia
2020-01-30
Cytheria
2023-05-13
Hanna
2023-05-13
Viveka
2023-05-13
Mora
2023-05-13
Bella
2023-05-13
Jacuqeline
2023-05-13
Tia
2023-05-13
Yelena
2023-05-13
Jayne
2023-05-13
Miette
2023-05-21
Nora
2023-05-31
Serafina
2023-06-12
Ursula
2023-07-20
Iris
2023-07-21
Isoke
2023-07-21