torch_optim_zero_grad Subroutine

public subroutine torch_optim_zero_grad(optim)

Zero Gradients on tensors associated with a Torch optimizer

Type Bound

torch_optim

Arguments

Type IntentOptional Attributes Name
class(torch_optim), intent(in) :: optim

Optimizer to zero gradients for

Source Code

  subroutine torch_optim_zero_grad(optim)
    class(torch_optim), intent(in) :: optim  !! Optimizer to zero gradients for

    interface
      subroutine torch_optim_zero_grad_c(optim_c) &
          bind(c, name = 'torch_optim_zero_grad')
        use, intrinsic :: iso_c_binding, only : c_ptr
        implicit none
        type(c_ptr), value, intent(in) :: optim_c
      end subroutine torch_optim_zero_grad_c
    end interface

    call torch_optim_zero_grad_c(optim%p)
  end subroutine torch_optim_zero_grad