def mlp()

in src/model.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

def mlp(x, scope, n_state, *, hparams):
    with tf.variable_scope(scope):
        nx = x.shape[-1].value
        h = gelu(conv1d(x, 'c_fc', n_state))
        h2 = conv1d(h, 'c_proj', nx)
        return h2