logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

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]

#python#深度学习
到底了