789d65f38f0949a3f03f61908a207cb558746769,stellargraph/layer/preprocessing_layer.py,GraphPreProcessingLayer,call,#GraphPreProcessingLayer#Any#,48

Before Change

                adj_T, tf.where(adj_T > adj, tf.ones_like(adj), tf.zeros_like(adj))
            )
            - tf.multiply(
                adj, tf.where(adj_T > adj, tf.ones_like(adj), tf.zeros_like(adj))
            )
        )
        // Add self loops.
        adj = adj + tf.linalg.diag(tf.ones(adj.shape[0]) - tf.diag_part(adj))
        // Normalization
        rowsum = tf.reduce_sum(adj, 1)
        d_mat_inv_sqrt = tf.diag(tf.rsqrt(rowsum))
        adj_normalized = tf.matmul(tf.matmul(d_mat_inv_sqrt, adj), d_mat_inv_sqrt)
        return adj_normalized

After Change

            // Normalization
            rowsum = tf.reduce_sum(adj, 1)
            d_mat_inv_sqrt = tf.linalg.diag(tf.math.rsqrt(rowsum))
            adj_normalized = tf.matmul(tf.matmul(d_mat_inv_sqrt, adj), d_mat_inv_sqrt)
            return adj_normalized

Instances