15bc4baff9c2dfa2e42947a417e45a159ab50dad,ludwig/models/modules/embedding_modules.py,EmbedSparse,__call__,#EmbedSparse#Any#Any#Any#Any#,287

Before Change

            self.initializer
        )

        multiple_hot_indexes = tf.multiply(
            input_sparse,
            tf.constant(np.array([range(len(self.vocab))], dtype=np.int32))
        )

        idx = tf.where(tf.not_equal(multiple_hot_indexes, 0))

        sparse_multiple_hot_indexes = tf.SparseTensor(
            idx,
            tf.gather_nd(multiple_hot_indexes, idx),
            tf.shape(multiple_hot_indexes, out_type=tf.int64)
        )

After Change

        sparse_multiple_hot_indexes = tf.SparseTensor(
            idx,
            idx[:, 1],
            tf.shape(input_sparse, out_type=tf.int64)
        )

Instances