Python PyTorch kthvalue用法及代碼示例

本文簡要介紹python語言中 torch.kthvalue 的用法。

用法:

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

參數：

• input(Tensor) -輸入張量。

• k(int) -k 為 k-th 最小元素

• dim(int,可選的) -找到第 k 個值的維度

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

關鍵字參數：

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

返回一個命名元組 (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 th 值時，此函數可能會不確定地為其中任何一個返回 indices。

例子：

>>> 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]]))