tf.argmax(vector, 1):返回的是vector中的最大值的索引號,如果vector是乙個向量,那就返回乙個值,如果是乙個矩陣,那就返回乙個向量,這個向量的每乙個維度都是相對應矩陣行的最大值元素的索引號。
reduce_mean(input_tensor,
axis=none)
第乙個引數input_tensor: 輸入的待降維的tensor;
第二個引數axis: 指定的軸,如果不指定,則計算所有元素的均值;
對於第二個引數axis,引數設為幾,那麼就是對第幾個維度進行降維求均值,其他維度數值保持不變。
import tensorflow as tf
x = [[[1, 2], [2, 3]], [[3, 4], [4, 5]], [[5, 6], [6, 7]]] #(3,2,2)的向量
xx = tf.cast(x, tf.float32)
mean_all = tf.reduce_mean(xx)
mean_0 = tf.reduce_mean(xx, axis=0)
mean_1 = tf.reduce_mean(xx, axis=1)
mean_2 = tf.reduce_mean(xx, axis=2)
with tf.session() as sess:
m_a, m_0, m_1, m_2 = sess.run([mean_all, mean_0, mean_1, mean_2])
print(m_a)
print(m_0) # (2,2)的向量
print(m_1) # (3,2)的向量
print(m_2) # (3,2)的向量
tensorflow學習筆記
tensorflow安裝可以直接通過命令列或者原始碼安裝,在此介紹tensorflow8命令列安裝如下 安裝tensorflow sudo pip install upgrade 另外,解除安裝tensorflow命令為 sudo pip uninstall tensorflow tensorflo...
Tensorflow學習筆記
1.如何在虛擬機器中安裝tensor flow 1 首先安裝pip pip install 2 pip install 2.學習tensorflow需要學習 python and linux 3.使用 tensorflow,你必須明白 tensorflow 1 使用圖 graph 來表示計算任務.2...
TensorFlow學習筆記
1 擬合直線 import the library import tensorflow as tf import numpy as np prepare train data train x np.linspace 1,1,100 temp1 train x,temp2 train x.shape,...