How to scale tf.nn.embedding_lookup_sparse
问题 I'm trying to build a very large sparse model (e.g. LR if there is only one embedding layer), the input dimension can be as large as 100000000, and the sample is very sparse, the average number of non zero value is around 100. Since the weights is very large and we have to partition and distribute it onto different servers. Here is the code: weights = tf.get_variable("weights", weights_shape, partitioner=tf.fixed_size_partitioner(num_shards, axis=0), initializer=tf.truncated_normal