The input is one or more feature columns, and args are embedding output size, to produce the results of `tf.nn.embedding_lookup()`.