torch.norm function is trying to work on dimension 1 of your input (the third argument = 1). But you input tensor has only 0th dimension (6 elements on 0th dimension). That is why the error appears. As I mention, it is complicated function and I don’t have time to inspect it, unfortunately. I suggest trace operations done on rs_center-6*conf[‘c0’] tensor back and find where it looses the dimension it has to have.