66866187ee32e7ec2f1e93cbd39a21a0afd1ccd3,ludwig/models/modules/embedding_modules.py,EmbedSparse,call,#EmbedSparse#Any#Any#Any#,265

Before Change

            self.dropout = None

    def call(self, inputs, training=None, mask=None):
        multiple_hot_indexes = tf.multiply(
            inputs,
            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(

After Change

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

Instances