简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
torch.topk() 函数详解
作用: 返回 列表中最大的n个值例子1:m=torch.arange(0,10)print(m.topk(3))torch.return_types.topk(values=tensor([9, 8, 7]),indices=tensor([9, 8, 7]))例子2:pred = torch.tensor([[-0.5816, -0.3873, -1.0215, -1.0145, 0.4053]
到底了