change levels.unsqueeze_(0) to levels = levels.unsqueeze(0) solve the issue
change levels.unsqueeze_(0) to levels = levels.unsqueeze(0) solve the issue