Tensorflow 学习笔记

tensorflow

1. 设置tensorflow输出日志级别

import os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

2. tf.argmax()函数

A = [[1, 3, 4, 5, 6]]
B = [[1, 3, 4], [2, 4, 1]]
# tf.argmax(y_, 1) 取的是行的最大值对应的下标的索引
# tf.argmax(y_, 0) 取的是列的最大值对应的下标的索引
with tf.Session() as sess:
    print(sess.run(tf.argmax(A, 1)))
    print(sess.run(tf.argmax(B, 0)))

运行结果：

[4]
[1 1 0]

3. tf.reduce_mean() 函数

• 函数用于计算张量tensor沿着指定的数轴（tensor的某一维度）上的的平均值，主要用作降维或者计算tensor（图像）的平均值。

reduce_mean(input_tensor,
                axis=None,
                keep_dims=False,
                name=None,
                reduction_indices=None)

• 第一个参数input_tensor： 输入的待降维的tensor;
• 第二个参数axis： 指定的轴，如果不指定，则计算所有元素的均值;
• 第三个参数keep_dims：是否降维度，设置为True，输出的结果保持输入tensor的形状，设置为False，输出结果会降低维度;
• 第四个参数name： 操作的名称;
• 第五个参数 reduction_indices：在以前版本中用来指定轴，已弃用;

4.