I have a tensor of shape [5sx,5sy,5*sz] and I’d like to create a tensor with shape [sx,sy,sz] where the value of each element:
new_tensor[x,y,z] = mean(old_tensor[5x:5x+5,5y:5y+5,5z:5z+5])
Preferably in one instruction that’s fast (and followed by autograd)
Any ideas?
Thanks,
Ian