= 1024" x-on:close-sidebar="sidebar=window.innerWidth >= 1024 && true">
package torch
-
torch.core
-
-
torch.toplevel
Legend:
Library
Module
Module type
Parameter
Class
Class type
Library
Module
Module type
Parameter
Class
Class type
type t = unit Ctypes.ptr
val t : t Ctypes.typ
val adam :
(float ->
float ->
float ->
float ->
float ->
t Torch_generated.return)
Torch_generated.result
val rmsprop :
(float ->
float ->
float ->
float ->
float ->
int ->
t Torch_generated.return)
Torch_generated.result
val sgd :
(float ->
float ->
float ->
float ->
bool ->
t Torch_generated.return)
Torch_generated.result
val add_parameters :
(t ->
Tensor.t Ctypes_static.ptr ->
int ->
unit Torch_generated.return)
Torch_generated.result
val set_learning_rate :
(t -> float -> unit Torch_generated.return) Torch_generated.result
val set_momentum :
(t -> float -> unit Torch_generated.return) Torch_generated.result
val zero_grad : (t -> unit Torch_generated.return) Torch_generated.result
val step : (t -> unit Torch_generated.return) Torch_generated.result
val free : (t -> unit Torch_generated.return) Torch_generated.result