TensorFlow函数（六）初始值生成函数

2024-08-24 03:39:33

1.常量生成函数

tf.constant(value, dtype)

生成一个初始值为常量value的数组

value：指定的常量
dtype：数据类型

tf.zeros(shape, dtype)

生成一个形状为shape、初始值全为0的数组

tf.ones(shape, dtype)

生成一个形状为shape、初始值全为1的数组

2.初始化为正太分布

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

生成一组符合标准正态分布的数组

shape：数据形状
mean：正态分布的均值，默认值 0
stddev：正态分布的标准差，默认值 1
seed：随机种子，指定seed的值相同生成同样的数据
dtype：数据类型

tf.truncated_normal(shape, mean, stddev, seed, dtype)

生成一组符合截断正态分布的数组，如果随机生成的值偏离均值超过2个标准差，那么这个数将会被重新生成

3.初始化为均匀分布

tf.random_uniform(shape, minval, maxval, seed, dtype)

生成一组符合均匀分布的数组

shape：数据形状
minval：最小值
maxval：最大值
seed：随机数种子
dtype：数据类型

均匀分布生成的随机数并不是从小到大或者从大到小均匀分布的，这里均匀分布的意义是每次从一组服从均匀分布的数里边随机抽取一个数

最新文章

热门文章