ChatGPT - Channel-wise Attention in PyTorch