1. 程式人生 > >TensorFlow函式之tf.random_normal()

TensorFlow函式之tf.random_normal()

tf.trandom_normal()函式是生成正太分佈隨機值

此函式有別於tf.truncated_normal()正太函式,請參考本部落格關於tf.truncated_normal()函式的介紹

TensorFlow函式之tf.truncated_normal()


tf.random_normal()函式的格式為:

tf.random_normal(shape, mean, stddev, dtype, seed, name)

引數說明:

  1. shape:表示生成隨機數的維度
  2. mean:正太分佈的均值,預設為0
  3. stddev:正太分佈的標準差
  4. dtype:生成正太分佈資料的型別
  5. seed:一個整數,當設定之後,每次生成的隨機數都一樣
  6. name:正太分佈的名字

下邊簡單例子說明函式用法

import tensorflow as tf

v = tf.random_normal([2, 2], mean=0, stddev=0.1, dtype=tf.float32, seed=1, name='v')

sess = tf.Session()
print(sess.run(v))
sess.close()

輸出為:

[[ 0.00127519  0.04090711]
 [-0.04193133 -0.09104948]]