PyTorch torch.kthvalue 函数

PyTorch torch.kthvalue 函数 Pytorch torch 参考手册 torch.kthvalue 是 PyTorch 中用于返回第k小的值的函数。返回输入张量在指定维度上的第k个最小值及其索引。函数定义 torch.kthvalue(input, k, dim=None, keepdim=False) 使用示例实例 [mycode4 type='python'] impo..

更新于 2026/7/16

教程详情页默认 noindex。

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)

输出结果为：

tensor(2)
tensor(2)

Pytorch torch 参考手册