search

malllabiisc / CompGCN

ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks
Apache License 2.0
597 stars 107 forks source link

Just a suggestion on Fourier transformation APIs for users whose pytorch version >= 1.8. #35

Closed nxznm closed 2 years ago

nxznm commented 2 years ago

This line could be changed to return torch.fft.irfftn(torch.conj(torch.fft.rfftn(a, (-1))) * torch.fft.rfftn(b, (-1)), (-1)) if your pytorch version >= 1.8.