Hi, actually I have run into the problem when trying to replicate the code; when I try:
sigma = x.std(dim=(0,2,3), keepdim=True)
I receive the error std(): argument 'dim' must be int, not tuple ; any thoughts?