floor 向下
ceil 向上
trunc 取证
frac 小数部分
pytorch 常用基础函数
https://www.jianshu.com/p/d678c5e44a6b
ge是大于等于 gt是大于
le是小于等于 lt 是小于 都要每个元素相对应的挨个去比较
上面是选择在哪个dim上找最大的，下面是不改变原来的dim意思
a.topk(3,dim=1)
表示是最大的三个数值
torch.where(condition,x,y)
相关文章：
相似解决方案
热门标签