set_requires_grad Module Subroutine

  pure module subroutine set_requires_grad(this, requires_grad)
    !! Set the requires_grad flag
    implicit none
    class(array_type), intent(inout) :: this
    logical, intent(in) :: requires_grad

    this%requires_grad = requires_grad
  end subroutine set_requires_grad