【Python-Tensorflow】tf.truncated_normal()解析与使用

1 作用

作用是产生均值为mean,标准差为stddev的正态分布的张量

2 参数解析

tf.truncated_normal(shape, mean, stddev)

shape表示生成张量的维度 mean是均值 stddev是标准差

3 举例使用

import tensorflow as tf
import numpy as np
# 产生一个[32,1,8]的张量的正态分布
c = tf.truncated_normal([32, 1, 8], dtype=tf.float64, stddev=1e-1)  

with tf.Session() as sess:
    print(sess.run(c))