89326c618dcf420a79139034b02250a09335ba01,rasa/utils/tensorflow/layers.py,DotProductLoss,_random_indices,#DotProductLoss#Any#Any#,601

Before Change

        // create first random array of indices
        out1 = rand_idxs()  // (1, num_neg)

        return tf.nest.map_structure(
            tf.stop_gradient,
            tf.while_loop(
                cond,
                body,

After Change

        self, batch_size: tf.Tensor, total_candidates: tf.Tensor
    ) -> tf.Tensor:
        return tf.random.uniform(
            shape=(batch_size, self.num_neg), maxval=total_candidates, dtype=tf.int32
        )

    @staticmethod

Instances