PyTorch Forums
weight_theta