Hello the following code snippet gives the error in the title. Snippet works on cpu but fails on 2 GPU environment, I haven’t checked with a single GPU. Code particularly fails on w_theta_sin line.
def forward(self, x, a):
    with torch.no_grad():
        self.W_hat.div_(torch.norm(self.W_hat, dim=5, keepdim=True))
        self.W_theta.fmod_(math.pi)
    W_theta_sin = torch.sin(self.W_theta) + eps

RuntimeError: diff_view_meta->output_nr_ == 0 ASSERT FAILED at /opt/conda/conda-bld/pytorch_1556653215914/work/torch/csrc/autograd/variable.cpp:209, please report a bug to PyTorch.
Searched the same error, I know that there is an open issue, but I did not understand much of it tbh:) is there a workaround for this?
Thanks in advance.