在tensorflow中定义一个非常简单的偏置初始化器

时间:2017-09-04 11:39:47

标签: tensorflow initialization tf-slim

我需要为卷积层偏差编写一个简单的初始化器。我正在使用tf.slim,因此我可以在调用卷积层时指定初始化程序,like so.

我想用我自己的自定义函数替换biases_initializer=init_ops.zeros_initializer(),该函数只是将偏差初始化为给定的常量,例如:

`biases_initializer=custom_initializer(value)`

我可以在其中指定值,例如value = -5

有人能告诉我这是怎么做到的吗?我花了大约一个小时阅读existing initializers,但仍然不知道如何实现这个简单的功能。

1 个答案:

答案 0 :(得分:0)

我终于发现没有必要定义该函数,因为已经存在tf.constant_initializer。以上只需通过以下方式实现:

biases_initializer = tf.constant_initializer(value)