def _linear(t_in, n_out):
assert len(t_in.get_shape()) == 2
v_w = tf.get_variable(
"w",
shape=(t_in.get_shape()[1], n_out),
initializer=tf.uniform_unit_scaling_initializer(
factor=INIT_SCALE))
v_b = tf.get_variable(
"b",
shape=n_out,
initializer=tf.constant_initializer(0))
return tf.einsum("ij,jk->ik", t_in, v_w) + v_b
评论列表
文章目录