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