如何使用Python topk()函数实现求最大和最小值示例？

本文共计427个文字，预计阅读时间需要2分钟。

函数介绍：a.topk(k) 求a中的最大值或最小值，返回这两个值，一个是a中的值（最大或最小），另一个是这个值的索引。

代码示例：pythonimport torcha=torch.randn((3, 5))print(a.topk(1)) # 返回最大值及其索引print(a.topk(1, largest=False)) # 返回最小值及其索引

函数介绍

a.topk()求a中的最大值或最小值，返回两个值，一个是a中的值（最大或最小），一个是这个值的索引。