举个例子先:
我要得到一组概率的数据中最大值出现的位置,利用argmax()函数可以解决。
但是坑在于,numpy(以下简称np了。。。)和 tensorflow (以下简称tf )都有这个函数
然后要是直接用的tf.argmax()去处理,得到的就是个tensor(张量)save成图像就报错。。。。
改用np.argmax(),让输出结果的形式变成numpy,然后存成.npy 或者存成图像就都行了~~~
类似的错误还有很多(如tf.squeeze() 和 np.mean() 都可以用来压缩维度,但是得到的数据类型有差别)
所以要正确选择到底是用np的函数,还是tf的函数。