tensor.topk 以及tensor.argmax

tensor.topk 以及tensor.argmax

topk(self, k, dim=None, largest=True, sorted=True):,返回两个值,values与indices。

argmax(self, dim=None, keepdim=False): 返回Tensor

python 复制代码
'''
具体使用方法定义里写的很清楚,topk中largest=False返回最小值,sorted打乱原有的元素顺序。
argmax中keepdim=True:保持维数不变,默认会减少一维。
''' 

import torch

a = torch.randn(2, 2, 2)
a.argmax(1)

'''
tensor([[0, 1],
        [1, 0]])
'''

a.topk(k=1)
'''
torch.return_types.topk(
values=tensor([[[ 0.8766],
         [-0.1330]],

        [[ 1.5773],
         [ 0.8146]]]),
indices=tensor([[[0],
         [1]],

        [[1],
         [0]]]))
'''
a.topk(k=1).values
'''
tensor([[[ 0.8766],
         [-0.1330]],

        [[ 1.5773],
         [ 0.8146]]])
'''
a.topk(1).indices
'''
tensor([[[0],
         [1]],

        [[1],
         [0]]])
'''
相关推荐
booooooty8 分钟前
基于Spring AI Alibaba的多智能体RAG应用
java·人工智能·spring·多智能体·rag·spring ai·ai alibaba
PyAIExplorer36 分钟前
基于 OpenCV 的图像 ROI 切割实现
人工智能·opencv·计算机视觉
风口猪炒股指标38 分钟前
技术分析、超短线打板模式与情绪周期理论,在市场共识的形成、分歧、瓦解过程中缘起性空的理解
人工智能·博弈论·群体博弈·人生哲学·自我引导觉醒
ai_xiaogui1 小时前
一键部署AI工具!用AIStarter快速安装ComfyUI与Stable Diffusion
人工智能·stable diffusion·部署ai工具·ai应用市场教程·sd快速部署·comfyui一键安装
聚客AI2 小时前
Embedding进化论:从Word2Vec到OpenAI三代模型技术跃迁
人工智能·llm·掘金·日新计划
weixin_387545642 小时前
深入解析 AI Gateway:新一代智能流量控制中枢
人工智能·gateway
聽雨2373 小时前
03每日简报20250705
人工智能·社交电子·娱乐·传媒·媒体
二川bro3 小时前
飞算智造JavaAI:智能编程革命——AI重构Java开发新范式
java·人工智能·重构
acstdm3 小时前
DAY 48 CBAM注意力
人工智能·深度学习·机器学习
澪-sl3 小时前
基于CNN的人脸关键点检测
人工智能·深度学习·神经网络·计算机视觉·cnn·视觉检测·卷积神经网络