torch_optim_step Subroutine

public subroutine torch_optim_step(optim)

Step a Torch optimizer

Type Bound

torch_optim

Arguments

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

Optimizer to step


Source Code

  subroutine torch_optim_step(optim)
    class(torch_optim), intent(in) :: optim  !! Optimizer to step

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

    call torch_optim_step_c(optim%p)
  end subroutine torch_optim_step