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

Before Change

        

        // Build a symmetric adjacency matrix.
        adj_T = tf.transpose(adj)
        adj = (
            adj
            + tf.multiply(
                adj_T, tf.where(adj_T > adj, tf.ones_like(adj), tf.zeros_like(adj))

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