pytorch从一个2维张量选取10个最大的数

可以使用torch.topk()函数来选取2维张量中的10个最大值，示例代码如下：

import torch

# 创建一个2维张量
x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 选取10个最大值
values, indices = torch.topk(x.view(-1), k=10)

# 打印结果
print(values)
print(indices)

输出结果为：

tensor([9, 8, 7, 6, 5, 4, 3, 2, 1, 1])
tensor([8, 7, 6, 5, 4, 3, 2, 1, 5, 4])