hi,
Are there bugs in incoming edge torch.matmul(e[ : , 0], self.A[0, 1: ]) ?
Edge (0, 1), (0, 7), (0, 4) are from the hip of a human body, if using self.A[0, 1: ], the first row of A is omitted.
From the point of view of incoming, Vertices 1, 4, 7 would get the edge from vertex 0.
suppose that torch.matmul(e[ : , 0], B)
B = [0 I]
B is 16 x 17. I is 16 x 16 identity. 0 is 16 x 1 zero column.