def expand_tile(units, axis):
assert axis in (<NUM_LIT:1>, <NUM_LIT:2>)<EOL>n_time_steps = tf.shape(units)[<NUM_LIT:1>]<EOL>repetitions = [<NUM_LIT:1>, <NUM_LIT:1>, <NUM_LIT:1>, <NUM_LIT:1>]<EOL>repetitions[axis] = n_time_steps<EOL>return tf.tile(tf.expand_dims(units, axis), repetitions)<EOL>
Expand and tile tensor along given axis
    Args:
        units: tf tensor with dimensions [batch_size, time_steps, n_input_features]
        axis: axis along which expand and tile. Must be 1 or 2