torch topk

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

pytorch中文官網文檔：http://www.mamicode.com/info-detail-2217311.html

沿給定dim維度返回輸入張量input中?k?個最大值。

如果不指定dim，則默認為input的最后一維。

如果為largest為?False?，則返回最小的?k?個值。

返回一個元組?(values,indices)，其中indices是原始輸入張量input中測元素下標。

如果設定布爾值sorted?為_True_，將會確保返回的?k?個值被排序。

參數:

input (Tensor) – 輸入張量

k (int) – “top-k”中的k

dim (int, optional) – 排序的維

largest (bool, optional) – 布爾值，控制返回最大或最小值

sorted (bool, optional) – 布爾值，控制返回值是否排序

out (tuple, optional) – 可選輸出張量 (Tensor, LongTensor) output buffer

1維的可以直接取值，

import torch

anch_ious = torch.Tensor([[1, 2, 3], [4, 5, 6]]).view(-1)

neg_count=4

top_data,index= torch.topk(anch_ious, neg_count

版權聲明：本文內容由網絡用戶投稿，版權歸原作者所有，本站不擁有其著作權，亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容，請聯系我們jiasou666@gmail.com 處理，核實后本網站將在24小時內刪除侵權內容。

版權聲明：本文內容由網絡用戶投稿，版權歸原作者所有，本站不擁有其著作權，亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容，請聯系我們jiasou666@gmail.com 處理，核實后本網站將在24小時內刪除侵權內容。