145170ca9bbd89aa01d8a40841e3c039d3683af8,stellargraph/layer/graph_attention.py,GraphAttention,call,#GraphAttention#Any#,211

Before Change

            N = K.int_shape(inputs[1])[-1]
            if N is not None:
                // create self-loops
                A = tf.linalg.set_diag(A, K.cast(np.ones((N,)), dtype="float"))
            else:
                raise ValueError(
                    "{}: need to know number of nodes to add self-loops; obtained None instead".format(

After Change

        out_indices = inputs[1]  // output indices (1 x K)
        A = inputs[2]  // Adjacency matrix (1 x N x N)

        print(">>", X.shape, A.shape, out_indices.shape)

        batch_dim, n_nodes, _ = K.int_shape(X)
        if batch_dim != 1:
            raise ValueError(

Instances