torch 0.14 · OCaml Package

torch
- Torch
  - Checkpointing
  - Cifar_helper
  - Cuda
  - Dataset_helper
  - Device
  - Ivalue
  - Layer
    
    Gru
    
    Lstm
  - Mnist_helper
  - Module
  - Optimizer
    
    Clip_grad
    
    Linear_interpolation
  - Scalar
  - Serialize
  - Tensor
  - Text_helper
  - Var_store
    
    Init
    
    Tensor_id
torch.core
- Torch_core
  - Device
  - Kind
  - Reduction
  - Torch_bindings
    
    C
    
    Cuda
    
    F
    
    Ivalue
    
    Module
    
    Optimizer
    
    Scalar
    
    Serialize
    
    Tensor
    
    TensorG
  - Torch_bindings_generated
    
    C
    
    F
    
    C0
    
    F
    
    C1
    
    F
    
    C10
    
    F
    
    C11
    
    F
    
    C12
    
    F
    
    C13
    
    F
    
    C14
    
    F
    
    C15
    
    F
    
    C16
    
    F
    
    C17
    
    F
    
    C18
    
    F
    
    C2
    
    F
    
    C3
    
    F
    
    C4
    
    F
    
    C5
    
    F
    
    C6
    
    F
    
    C7
    
    F
    
    C8
    
    F
    
    C9
    
    F
  - Torch_generated
  - Wrapper
    
    Cuda
    
    Ivalue
    
    Tag
    
    Module
    
    Optimizer
    
    Scalar
    
    Serialize
    
    Tensor
  - Wrapper_generated
    
    C
    
    Cuda
    
    Ivalue
    
    Module
    
    Optimizer
    
    Scalar
    
    Serialize
    
    Tensor
    
    TensorG
  - Wrapper_generated_intf
    
    S
torch.toplevel
- Torch_toplevel
torch.vision
- Torch_vision
  - Alexnet
  - Densenet
  - Efficientnet
  - Image
    
    Loader
  - Imagenet
    
    Classes
    
    Loader
  - Inception
  - Mobilenet
  - Resnet
  - Squeezenet
  - Stb_image
  - Stb_image_write
  - Vgg

Legend:
Library
Module
Module type
Parameter
Class
Class type

include module type of Torch_core.Wrapper.Module

type t

val load : string -> t

val forward : 
  t ->
  Torch_core.Wrapper.Tensor.t list ->
  Torch_core.Wrapper.Tensor.t

val forward_ : t -> Ivalue.t list -> Ivalue.t