pytorch api(2)

floor 向下 ceil 向上 trunc 取證 frac 小數部分 pytorch 常用基礎函數 https://www.jianshu.com/p/d678c5e44a6b ge是大於等於 gt是大於 le是小於等於 lt 是小於 都要每個元素相對應的挨個去比較 上面是選擇在哪個dim上找最大的,下面是不改變原來的dim意思 a.topk(3,dim=1) 表示是最大的三個數值 torch.
相關文章
相關標籤/搜索