package torch

  1. Overview
  2. Docs

Module Torch.LayerSource

Sourcetype t
Sourcetype t_with_training
Sourceval set_training : t_with_training -> is_training:Base.bool -> t
Sourceval with_training : t -> t_with_training
Sourcetype activation =
  1. | Relu
  2. | Softmax
  3. | Log_softmax
  4. | Tanh
  5. | Leaky_relu
  6. | Sigmoid
Sourceval linear : Var_store.t -> ?activation:activation -> ?use_bias:Base.bool -> ?w_init:Var_store.Init.t -> input_dim:Base.int -> Base.int -> t
Sourceval conv2d : Var_store.t -> ksize:(Base.int * Base.int) -> stride:(Base.int * Base.int) -> ?activation:activation -> ?use_bias:Base.bool -> ?w_init:Var_store.Init.t -> ?padding:(Base.int * Base.int) -> ?groups:Base.int -> input_dim:Base.int -> Base.int -> t
Sourceval conv2d_ : Var_store.t -> ksize:Base.int -> stride:Base.int -> ?activation:activation -> ?use_bias:Base.bool -> ?w_init:Var_store.Init.t -> ?padding:Base.int -> ?groups:Base.int -> input_dim:Base.int -> Base.int -> t
Sourceval conv_transpose2d : Var_store.t -> ksize:(Base.int * Base.int) -> stride:(Base.int * Base.int) -> ?activation:activation -> ?use_bias:Base.bool -> ?w_init:Var_store.Init.t -> ?padding:(Base.int * Base.int) -> ?output_padding:(Base.int * Base.int) -> ?groups:Base.int -> input_dim:Base.int -> Base.int -> t
Sourceval conv_transpose2d_ : Var_store.t -> ksize:Base.int -> stride:Base.int -> ?activation:activation -> ?use_bias:Base.bool -> ?w_init:Var_store.Init.t -> ?padding:Base.int -> ?output_padding:Base.int -> ?groups:Base.int -> input_dim:Base.int -> Base.int -> t
Sourceval batch_norm2d : Var_store.t -> ?w_init:Var_store.Init.t -> ?cudnn_enabled:Base.bool -> ?eps:Base.float -> ?momentum:Base.float -> Base.int -> t_with_training
Sourceval id : t
Sourceval of_fn : (Tensor.t -> Tensor.t) -> t
Sourceval of_fn_ : (Tensor.t -> is_training:Base.bool -> Tensor.t) -> t_with_training
Sourceval fold : t Base.list -> t
Sourceval apply : t -> Tensor.t -> Tensor.t
Sourceval apply_ : t_with_training -> Tensor.t -> is_training:Base.bool -> Tensor.t
Sourcemodule Lstm : sig ... end
OCaml

Innovation. Community. Security.