PyTorch torch.kthvalue 函数
Pytorch torch 参考手册torch.kthvalue 是 PyTorch 中用于返回第k小的值的函数。返回输入张量在指定维度上的第k个最小值及其索引。
函数定义
torch.kthvalue(input, k, dim=None, keepdim=False)
使用示例
实例
import torch
# 创建张量
x = torch.tensor([3, 1, 2, 4])
# 返回第2小的值和索引
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
# 创建张量
x = torch.tensor([3, 1, 2, 4])
# 返回第2小的值和索引
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
输出结果为：
tensor(2) tensor(2)