397f52d1e457b421b99ae352d0284ada00b321d8,stellargraph/layer/sort_pooling.py,SortPooling,call,#SortPooling#Any#,68

Before Change

        // Truncate or pad to size self.k
        if outputs.shape[1] < self.k:
            outputs = tf.pad(outputs, [[0, 0], [0, self.k - outputs.shape[1]], [0, 0]])
        elif outputs.shape[1] > self.k:
            outputs = outputs[:, : self.k, :]

After Change

            inputs,
        )

        outputs_shape = tf.shape(outputs)

        outputs = tf.cond(
            tf.math.less(outputs_shape, self.k)[1],
            true_fn=lambda: tf.pad(
                outputs, [[0, 0], [0, (self.k - outputs_shape)[1]], [0, 0]]
            ),

Instances