torch.kthvalue

torch.kthvalue(input, k, dim=None, keepdim=False, *, out=None)

傳回一個 namedtuple (values, indices)，其中 values 是給定維度 dim 中 input 張量每一列的第 k 個最小元素。而 indices 是找到的每個元素的索引位置。

如果未給定 dim，則會選擇 input 的最後一個維度。

如果 keepdim 為 True，則 values 和 indices 張量的尺寸與 input 相同，但維度 dim 的尺寸為 1。 否則，dim 會被壓縮 (參見 torch.squeeze())，導致 values 和 indices 張量的維度都比 input 張量少 1。

注意

當 input 是一個 CUDA 張量，且有多個有效的第 k 個值時，此函數可能會不確定性地返回其中任何一個值的 indices。

參數

• input (Tensor) – 輸入張量。

• k (int) – 第 k 個最小元素的 k 值

• dim (int, optional) – 尋找第 k 個值的維度

• keepdim (bool) – 輸出張量是否保留 dim。

關鍵字參數

out (tuple, optional) – 可選擇給定 (Tensor, LongTensor) 的輸出元組以用作輸出緩衝區

範例

>>> x = torch.arange(1., 6.)
>>> x
tensor([ 1.,  2.,  3.,  4.,  5.])
>>> torch.kthvalue(x, 4)
torch.return_types.kthvalue(values=tensor(4.), indices=tensor(3))

>>> x=torch.arange(1.,7.).resize_(2,3)
>>> x
tensor([[ 1.,  2.,  3.],
        [ 4.,  5.,  6.]])
>>> torch.kthvalue(x, 2, 0, True)
torch.return_types.kthvalue(values=tensor([[4., 5., 6.]]), indices=tensor([[1, 1, 1]]))