torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor). 主要用途:依照大小,从矩阵某维度取值和取索引。
確定! 回上一頁