Hi @KFrank!
Thanks for the response! So, it seems that it should be ok when using it to unsqueeze new dims at the end dimensions but not any in between? (this is the special use case of it?) For example, x[:, None] is fine by x[:, None, :] isn’t?
I’m only using it at the moment in order to broadcast a batch of scalars over a batch of matrices. For example,
A = torch.randn(100, 2, 4, 4) #some matrcies
scalars = torch.randn(100, 2) #scalar scalars
scaled_A = scalar[:,:,None,None] * A

Perhaps it might be safer to do this?
A = torch.randn(100, 2, 4, 4) #some matrcies
scalars = torch.randn(100, 2) #scalar scalars
scaled_A = scalar.unsqueeze(2).unsqueeze(3) * A

Thank you for the help!