def _build(self):
w_init = tf.random_normal(
stddev=0.01,
shape=(
self._filter_height,
self._filter_width,
self._input_depth,
self._output_depth
),
dtype=D_TYPE,
name='w_init'
)
b_init = tf.zeros(
shape=(self._output_depth,),
dtype=D_TYPE,
name='b_init'
)
self._w = tf.Variable(w_init, dtype=D_TYPE, name='w')
self._b = tf.Variable(b_init, dtype=D_TYPE, name='b')
评论列表
文章目录