Hi all,
Assuming I have a input tensor with shape B x N x C x H x W, I want to apply the same CNN model f across the dimension N.
For now, I use for-loop to go over it.
out = []
for i in range(N):
    out.append(self.conv(input[:, i])

I was wondering if there is any elegant way to do this.
Thanks.