torch中如何找出矩陣中元素之大於某個閾值的所有元素的下標？torch.find() ?

近期用到了torch中要查找矩陣中元素大於某個閾值的函數，torch中的函數一般都爲 torch.函數名  比如 torch.max(), 於是乎，搜了torch.find() ,各種搜索都搜不到, 幾經周折終於搞定，用法如下：a=torch.Tensor(2,3):range(1,6); 想找到元素值大於3的元素下標，則下標索引 b=a[a:gt(3)]  ，，， 錯了錯了，返回的是元素的值，，

>>阅读原文<<