torch_model_print_parameters Subroutine

public subroutine torch_model_print_parameters(self)

Prints the parameters associated with a model NOTE: While viewing parameters in this way can be helpful for small toy models, it will produce large amounts of output for models with many, large, or high-dimensional parameters. In particular, tensors of 3 or more dimensions will be represented in terms of 2D arrays.

Type Bound

torch_model

Arguments

Type IntentOptional Attributes Name
class(torch_model), intent(in) :: self

Model to print the parameters of

Source Code

  subroutine torch_model_print_parameters(self)
    class(torch_model), intent(in) :: self  !! Model to print the parameters of

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

    call torch_jit_model_print_parameters_c(self%p)
  end subroutine torch_model_print_parameters