Shape of Image [batch_size, num_channels, width, height]

[gurdeepmaurya]

Hi, Just wanted to confirm if shape of image is [batch_size, num_channels, width, height],

usually it's [batch_size, num_channels, height, width].

https://github.com/evelinehong/slot-attention-pytorch/blob/1518c23c312f93b961f8c071136f0093b741863a/model.py#L176

Thanks and regards