index 1 is out of bounds for dimension 1 with size 1

BUAABIGSCity / PDFormer

[AAAI2023] A PyTorch implementation of PDFormer: Propagation Delay-aware Dynamic Long-range Transformer for Traffic Flow Prediction.

MIT License

209 stars 35 forks source link

Closed aligaduocaisang closed 2 months ago

aligaduocaisang commented 2 months ago

pattern_key_list.append(self.pattern_embeddings[i](self.pattern_keys[..., i]).unsqueeze(-1))

IndexError: index 1 is out of bounds for dimension 1 with size 1 请问这是什么情况