torch_optim Derived Type

type, public :: torch_optim

Type for holding a torch optimizer.


Inherits

type~~torch_optim~~InheritsGraph type~torch_optim torch_optim c_ptr c_ptr type~torch_optim->c_ptr p

Components

Type Visibility Attributes Name Initial
type(c_ptr), public :: p = c_null_ptr

pointer to the optimizer in memory

Finalization Procedures

final :: torch_optim_delete

  • public subroutine torch_optim_delete(optim)

    Deallocate a Torch optimizer

    Arguments

    Type IntentOptional Attributes Name
    type(torch_optim), intent(inout) :: optim

    Optimizer to deallocate

Type-Bound Procedures

procedure, public :: step => torch_optim_step

  • public subroutine torch_optim_step(optim)

    Step a Torch optimizer

    Arguments

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

    Optimizer to step

procedure, public :: zero_grad => torch_optim_zero_grad

  • public subroutine torch_optim_zero_grad(optim)

    Zero Gradients on tensors associated with a Torch optimizer

    Arguments

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

    Optimizer to zero gradients for

Source Code

  type torch_optim
    type(c_ptr) :: p = c_null_ptr  !! pointer to the optimizer in memory
  contains
    procedure :: step => torch_optim_step
    procedure :: zero_grad => torch_optim_zero_grad
    final :: torch_optim_delete
  end type torch_optim