首页是torch不是numpy

是torch不是numpy

时间: 2023-10-30 16:04:49 浏览: 77

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.

好的，那么你可以使用以下代码来获取想要的数： ```python import torch A = torch.rand(10, 10) # 随机生成一个10*10的矩阵 index = torch.tensor([2, 5, 8, 1, 4, 7, 0, 3, 6, 9]) # 索引张量 result = A[torch.arange(10), index] # 获取想要的数 print(result) ``` 这里同样用到了PyTorch的高级索引，其中`torch.arange(10)`生成了一个0到9的张量，表示矩阵A的行索引，`index`表示列索引。通过`A[torch.arange(10), index]`即可获取想要的结果。

阅读全文