torch 取一个张量中存储在另一个张量中的序号对应值
时间: 2024-05-15 12:14:02 浏览: 96
抓取儲存卡的序號
4星 · 用户满意度95%
可以使用 PyTorch 中的 `index_select` 函数来实现这个功能。`index_select` 函数接收两个参数:一个是需要取出值的张量,另一个是存储序号的张量。
以下是一个示例代码:
```python
import torch
# 创建一个需要取值的张量
values = torch.Tensor([1, 2, 3, 4, 5])
# 创建一个存储序号的张量
indices = torch.LongTensor([0, 2, 4])
# 使用 index_select 函数取出需要的值
result = torch.index_select(values, 0, indices)
print(result)
```
输出结果为:
```
tensor([1., 3., 5.])
```
其中,`torch.index_select(values, 0, indices)` 表示从 `values` 张量中取出第 0 维度上序号为 `indices` 的元素,并返回一个新的张量。
阅读全文