首页pytorch将tensor转化为numpy

pytorch将tensor转化为numpy

时间: 2023-06-05 15:47:24 浏览: 127

pytorch 实现tensor与numpy数组转换

看代码，tensor转numpy: a = torch.ones(2,2) b = a.numpy() c=np.array(a) #也可以转numpy数组 print(type(a)) print(type(b)) print(a) print(b) 输出为： <class> <class> tensor([[1., 1.], [1., 1.]]) [[1. 1.] [1. 1.]] numpy转tensor： import torch import numpy as np a = np.ones(5) b = torch.

可以使用`.numpy()`方法将PyTorch Tensor转换为numpy数组。示例: ``` import torch import numpy as np tensor = torch.randn(3, 3) numpy_array = tensor.numpy() print(type(numpy_array)) # <class 'numpy.ndarray'> ```

阅读全文