python取tensor数组中前10大的数

假设你使用的是 PyTorch，可以通过以下代码取得一个 tensor 数组中前 10 大的数：

import torch

# 创建一个 tensor 数组
tensor_array = torch.tensor([1, 5, 3, 2, 4, 6, 8, 9, 7, 10])

# 取前 10 大的数
top_k_values, top_k_indices = torch.topk(tensor_array, k=10)

# 输出结果
print(top_k_values)  # tensor([10,  9,  8,  6,  5,  4,  3,  2,  1])
print(top_k_indices)  # tensor([9, 7, 6, 5, 1, 4, 2, 3, 0])

其中，`torch.topk()` 函数的第一个参数是待排序的 tensor 数组，第二个参数 `k` 表示需要取前几个数。函数返回两个值，第一个值是前 k 大的数组成的 tensor 数组，第二个值是这些数在原 tensor 数组中的下标组成的 tensor 数组。