会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Tomorrow
博客园
首页
新随笔
联系
订阅
管理
2022年3月24日
torch topk函数
摘要: 这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index。 用法 torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor) input:一个tens
阅读全文
posted @ 2022-03-24 10:37 Tomorrow1126
阅读(2459)
评论(0)
推荐(0)
编辑
公告