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.
相关文章
相关标签/搜索