目录
向量范数
tf.norm()求范数
在指定维度上求范数 (比如在行方向求二范数,每行进行求解,0为列,1为行)
求1范数,ord=1
求极值平均值
reduce_min/max/mean
不指定维度则求所有值的最大最小平均
求指定维度的均值,指定axis,则求每一行的的最大最小平均值
排序及返回极值的位置
sort排序,direction=descending降序,排序完后为43210
argsort:返回最大值所在的位置,最大的是4,所在位置为3,次大值为3,所在位置为2~
gather:收集,按照argsort坐标收集,得到43210
二维
topk
例子,最大的前两个,tf.top_k.indices
topk accuracy