tensorflow相關tensor計算函式

1. tf.split

該函式主要用於對tensor進行分割，一般在設定多gpu平行計算時經常會被用到，主要是將乙個batch資料集進行平分，分配給各個gpu，最後再彙總各個gpu得到的損失，從而加快模型的訓練速度，其主要引數的定義如下：

import tensorflow as tf

tf.split(
value, 
num_or_size_splits, 
axis=0, 
num=none, 
name="split"
)

該函式主要是對輸入的tensor列表中每乙個tensor進行加總，要求每個tensor的維度必須相同，當開啟平行計算時，該函式也經常被用來計算各個gpu得到的損失，其主要引數定義如下：

import tensorflow as tf

tf.add_n(inputs, name=none)

tensorflow相關函式 to

函式宣告 to categorical y,num classes none,dtype float32 作用 將整型標籤轉為onehot。y為int陣列，num classes為標籤類別總數，大於max y 標籤從0開始的 返回 如果num classes none，返回len y max y 1...

Tensorflow 安裝 使用 相關

最近在ubuntu中折騰tensorflow 花了兩晚上，終於搞定 1.安裝ubuntu 64位，64位，tensorflow是不支援32位的 2.安裝完後，安裝python 2.7 3.安裝pip 再upgrade 一下 4.安裝tensorflow 不用按照中文 那個pip 一大堆，直接pip ...

TensorFlow 梯度優化相關

tf.trainable variables可以得到整個模型中所有trainable true的variable，也是自由處理梯度的基礎 tf.gradients 用來計算導數。該函式的定義如下所示 def gradients ys,xs,grad ys none,name gradients co...