torch.topk¶. 火炬。 topk ( 输入,k,昏暗的=没有,最大= True,排序= True,*,= None) - >(张量,LongTensor ) ¶. 返回 k 给定的最大元素 输入 沿着给定维度的张 ...
確定! 回上一頁