https://stackoverflow.com/a/49134587
求argmax、加newaxis、转变data类型时尽量用tf自带的函数：
tf.argmax、[tf.newaxis, :]、tf.cast(data, datatype=tf.float32)
保持输出为tensor，不然之后转型很麻烦的…
相关文章：
相似解决方案
热门标签
https://stackoverflow.com/a/49134587
求argmax、加newaxis、转变data类型时尽量用tf自带的函数：
tf.argmax、[tf.newaxis, :]、tf.cast(data, datatype=tf.float32)
保持输出为tensor，不然之后转型很麻烦的…
相关文章：