tf.argmax tf2版本

对于tf.argmax,这个函数有点奇怪，axis=0指的是计算矩阵每列的最大值索引，axis=1计算行最大值索引
与numpy 相同

import tensorflow as tf
import numpy as npa=np.array([[2,4,5,7],[9,3,6,2]])
print('-'*30+'分割线'+'-'*30)
print(a)
print('-'*30+'分割线'+'-'*30)a1=tf.argmax(a,axis=0)print('tf.argmax(a,axis=0)=',a1)
print('-'*30+'分割线'+'-'*30)
a1=np.argmax(a,axis=0)
print('np.argmax(a,axis=0)=',a1)
print('-'*30+'分割线'+'-'*30)a1=tf.argmax(a,axis=1)print('tf.argmax(a,axis=1)=',a1)
print('-'*30+'分割线'+'-'*30)a1=np.argmax(a,axis=1)print('np.argmax(a,axis=1)=',a1)
print('-'*30+'分割线'+'-'*30)

------------------------------分割线------------------------------
[[2 4 5 7][9 3 6 2]]
------------------------------分割线------------------------------
tf.argmax(a,axis=0)= tf.Tensor([1 0 1 0], shape=(4,), dtype=int64)
------------------------------分割线------------------------------
np.argmax(a,axis=0)= [1 0 1 0]
------------------------------分割线------------------------------
tf.argmax(a,axis=1)= tf.Tensor([3 0], shape=(2,), dtype=int64)
------------------------------分割线------------------------------
np.argmax(a,axis=1)= [3 0]
------------------------------分割线------------------------------

tf.argmax tf2版本相关推荐

ROS专题----tf和tf2坐标变换
ROS专题----tf和tf2坐标变换 ---- 工作区设置如果您尚未创建用于完成教程的工作区, 请单击此处查看一些简要说明 . 从tf1迁移到tf2 转换数据类型这是对转换数据类型的语法更改的快 ...
pythonargmaxaxis1_np.argmax(input,axis)和tf.argmax(input,axis)使用
np.argmax(input,axis)和tf.argmax(input,axis)分别是numpy和TensorFlow底下的求最大值索引的方法,用法基本一致,只有默认情况下有细微差别,以及传入的 ...
tensorflow基础（2）-tf.argmax和axis
tf.argmax() 这个函数的主要功能是返回最大值所在的坐标.主要用在分类的时候,如果只是简单的输出,只是对于每一类可能性的预测的输出,但是我要要的输出必须是确定的哪一类,所以需要确定里面的最大的 ...
tensorflow笔记：tf.argmax()和tf.equal()的使用
原文:https://blog.csdn.net/abc13526222160/article/details/85299901 1.tf.argmax(vector, 1) 返回的是vector中的 ...
pointnet train 第十句correct = tf.equal(tf.argmax(pred, 1), tf.to_int64(labels_pl))
correct = tf.equal(tf.argmax(pred, 1), tf.to_int64(labels_pl)) 首先看tf.argmax(pred,1)因为pred,是32,40的ten ...
【Tensorflow】 tf.equal(tf.argmax(y, 1),tf.argmax(y_, 1))用法
[Tensorflow] tf.equal(tf.argmax(y, 1),tf.argmax(y_, 1))用法作用:输出正确的预测结果利用tf.argmax()按行求出真实值y_.预测值y最大 ...
『tensorflow笔记』tf.argmax()和tf.equal()的使用
1.tf.argmax(vector, 1) 返回的是vector中的最大值的索引号,如果vector是一个向量,那就返回一个值,如果是一个矩阵,那就返回一个向量,这个向量的每一个维度都是相对应矩阵行 ...
Tensorflow中的tf.argmax()函数
转载出处:http://www.cnblogs.com/willnote/p/6758953.html 官方API定义 tf.argmax(input, axis=None, name=None, d ...
tf.argmax（）的用法
tf.argmax()的主要功能是找出最大值并返回索引,注意是索引.下面是主要的内参数 tf.argmax(input,axis=None,name=None,dimension=None,outpu ...

tf.argmax tf2版本

tf.argmax tf2版本相关推荐

最新文章

热门文章