def _build(self):
beta_init = tf.zeros(
shape=self._size,
dtype=D_TYPE
)
gamma_init = tf.ones(
shape=self._size,
dtype=D_TYPE
)
self._beta = tf.Variable(
name='beta',
initial_value=beta_init,
dtype=D_TYPE
)
self._gamma = tf.Variable(
name='gamma',
initial_value=gamma_init,
dtype=D_TYPE
)
评论列表
文章目录