Efficient training of models with large number of input features (>10k)

[achoum]

Background

Currently, the training graph contains one tf op for each input features. In case of large number of features (or in the case of multi dimensional features), this can lead to a large overhead (large memory consumption, large training initialization stage).

Features request

Support for multi dimensional features without creating an op for each dimension.