zouni

常用函数

unstack

作用：把向量按照维度拆分

gather

通过索引获取 标量或者向量

gather_nd

可以索引多维

argmax 最大值的索引

tf.argmax([1,2,3,4,5,6,7,8])

<tf.Tensor: shape=(), dtype=int64, numpy=7>

softmax 概率最大的索引