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,
                loop_vars=[idx1, out1],
                shape_invariants=[idx1.shape, tf.TensorShape([None, self.num_neg])],
                parallel_iterations=self.parallel_iterations,
            ),
        )[1]

    @staticmethod
    def _sample_idxs(batch_size: tf.Tensor, x: tf.Tensor, idxs: tf.Tensor) -> tf.Tensor:
        Sample negative examples for given indices

After Change

    def _random_indices(
        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
    def _sample_idxs(batch_size: tf.Tensor, x: tf.Tensor, idxs: tf.Tensor) -> tf.Tensor:
        Sample negative examples for given indices

Instances