Tensorflow學習筆記

2021-09-02 15:05:10 字數 870 閱讀 9518

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,...