工资「喂饱肚子」，副业「养活灵魂」！
欢迎大家交换友链，可在https://www.stubbornhuang.com/申请友情链接进行友链交换申请！
如果觉得本站的内容有帮助，可以考虑打赏博主哦！
本站会放置Google广告用于维持域名以及网站服务器费用。
问题反馈可发送邮件到stubbornhuang@qq.com
在本站开通年度VIP，无限制下载本站资源和阅读本站文章
感谢大家访问本站，希望本站的内容可以帮助到大家！
计算机图形学与计算几何经典必备书单整理，下载链接可参考：https://www.stubbornhuang.com/1256/
本站由于前段时间遭受到大量临时和国外邮箱注册，所以对可注册的邮箱类型进行了限制！

Pytorch – torch.topk参数详解与使用

StubbornHuang Pytorch 发布于2022-12-15 阅读 3,087次 0次评论 0次点赞本文共910个字，阅读需要3分钟。

1 torch.topk

形式

torch.topk(input, k, dim=None, largest=True, sorted=True, *, out=None)

或者

Tensor.topk(k, dim=None, largest=True, sorted=True)

功能

返回输入张量input在指定dim上k个最大的元素。

如果没有指定dim，则默认选择最后一个维度。

如果largest为false，则返回最小的元素。

参数

input(Tensor)：输入张量
k(int)：top-k中的k，k个最大的元素
dim：排序的维度
largest(bool)：如果为false，则返回最小的元素，默认为true
sorted(bool)：如果为true，则返回排序好的k个元素，默认为true

返回值

返回一个(value,indices)的元素，其中value为dim上最大k个元素的值，indices为dim上最大k个元素对应的索引

使用示例

import torch

if __name__ == '__main__':
    a = torch.randn(size=(4, 4))
    print(a)

    value, indices = torch.topk(a, 2)
    print("==========value============")
    print(value)

    print("==========indices============")
    print(indices)

输出

tensor([[ 0.8897, -0.1849, -0.0502,  0.6197],
        [-1.6369,  0.9490, -0.8909, -0.5208],
        [-1.2246,  1.5075,  0.6353, -0.3103],
        [-0.5476, -0.9881, -0.5171, -1.0863]])
==========value============
tensor([[ 0.8897,  0.6197],
        [ 0.9490, -0.5208],
        [ 1.5075,  0.6353],
        [-0.5171, -0.5476]])
==========indices============
tensor([[0, 3],
        [1, 3],
        [1, 2],
        [2, 0]])

欢迎扫码关注我的微信公众号，及时获取文章更新