Is there an issue with the description of the shape of the returned tensor H?

[sonrisa07]

pytorch_geometric_temporal/torch_geometric_temporal/nn/attention/stgcn.py

Lines 34 to 37 in b2833be

	Return types:
	* **H** (torch.FloatTensor) - Output data of shape
	(batch_size, in_channels, num_nodes, input_time_steps).
	"""

Initially, X.permute(0, 3, 2, 1) is applied, and again at the end. Aren't the dimensions still consistent with the input shape: (batch_size, input_time_steps, num_nodes, in_channels)?