torch.topk ... Returns the k largest elements of the given input tensor along a given dimension. If dim is not given, the last dimension of the input is chosen.
確定! 回上一頁