Fixing RuntimeError: Element 0 of Tensors Does Not Require grad and Does Not Have a grad_fn
RuntimeError: Element 0 of Tensors Does Not Require Grad and Does Not Have a Grad_fn
RuntimeError: Element 0 of Tensors Does Not Require Grad and Does Not Have a Grad_fn