torch.Tensor&numpy.ndarray

最新推荐文章于 2023-08-01 15:51:15 发布

会编程的漂亮小姐姐

最新推荐文章于 2023-08-01 15:51:15 发布

阅读量193

点赞数

分类专栏： Python AI

本文链接：https://blog.csdn.net/u014229742/article/details/117306351

版权

Python 同时被 2 个专栏收录

171 篇文章 2 订阅

订阅专栏

17 篇文章 0 订阅

订阅专栏

import torch
import numpy as np

a = torch.tensor([[1,2],[3,4]],dtype=torch.float32)
print(a,type(a),a.dtype)
b = np.array([[1,2],[3,4]],dtype=np.float32)
print(b,type(b),b.dtype)

c = a.numpy()   # <class 'torch.Tensor'>------><class 'numpy.ndarray'>
print(c,type(c))

d = torch.from_numpy(b) #<class 'numpy.ndarray'>------><class 'torch.Tensor'>
print(d,type(d),d.dtype)

"""
tensor([[1., 2.],
        [3., 4.]]) <class 'torch.Tensor'> torch.float32
[[1. 2.]
 [3. 4.]] <class 'numpy.ndarray'> float32
[[1. 2.]
 [3. 4.]] <class 'numpy.ndarray'>
tensor([[1., 2.],
        [3., 4.]]) <class 'torch.Tensor'> torch.float32
"""