PyG version: 2.0.4 PyTorch version: 1.11.0 OS: Ubuntu 20.04 Python version:3.8 CUDA/cuDNN version:11.3 How you installed PyTorch and PyG (conda,pip, source):pip Any other relevant information (e.g., version oftorch-scatter): GPU RTX 4090, torch-scatter:2.0.9, torch-sparse:0.6.15 ...
🐛 Bug It seems that the torch_geometric.nn.TransformerConv layer will give an output which is of size out_channels * heads by default rather than just out_channels. Though this only happens when concat=True, it's a bit awkward to chain m...