常用函数

unstack

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

unstack

gather

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

gather

gather_nd

可以索引多维

argmax 最大值的索引

softmax 概率最大的索引

最后更新于

这有帮助吗?