tf.constant_initializer

北木.

于 2020-02-29 21:29:24 发布

阅读量5.7k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： Tensorflow 文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/weixin_43283397/article/details/104582123

tf.constant_initializer()

参数：

value: Python标量、值列表或元组，或n维Numpy数组。初始化变量的所有元素将在value参数中设置为对应的值。

dtype: 数据类型。

verify_shape: 布尔值，用于验证value的形状。如果为真，如果value的形状与初始化张量的形状不兼容，初始化器将抛出错误。

1. 大小正好的初始化

import numpy as np
import tensorflow as tf
value = [0, 1, 2, 3, 4, 5, 6, 7]
# value = np.array(value) 
# value = value.reshape([2, 4])   效果一样
init = tf.constant_initializer(value)
 
print('fitting shape:')
with tf.Session():
    x = tf.get_variable('x', shape=[2, 4], initializer=init)
    x.initializer.run()