# Python程序查找张量的第K个元素
#导入所需的库
import torch

#创建一个1D张量
T = torch.Tensor([2.334,4.433,-4.33,-0.433,5, 4.443])
print("原始张量：\n", T)

#查找已排序张量中的第3个元素。首先按升序对张量进行排序，然后返回排序后张量中第K个元素的值和元素在原始张量中的索引
value，index = torch.kthvalue(T, 3)

#输出第3个元素的值和索引
print("第3个元素的值：", value)
print("第3个元素的索引：", index)

原始张量：
   tensor([ 2.3340, 4.4330, -4.3300, -0.4330, 5.0000, 4.4430])
第3个元素的值：tensor(2.3340)
第3个元素的索引：tensor(0)

# Python程序查找张量的前K个元素
#导入所需的库
import torch

#创建一个1D张量
T = torch.Tensor([2.334,4.433,-4.33,-0.433,5, 4.443])
print("原始张量：\n", T)

#查找张量中前k=2个或最大的2个元素，返回在原始张量中前2个最大元素的值及它们的索引
values，indices = torch.topk(T, 2)

#输出前2个元素的值和索引
print("前2个元素的值：", values)
print("前2个元素的索引：", indices)

原始张量：
   tensor([ 2.3340, 4.4330, -4.3300, -0.4330, 5.0000, 4.4430])
前2个元素的值：tensor([5.0000, 4.4430])
前2个元素的索引：tensor([4, 5])