class

Torch::Module

Inherits Reference < Object

Constructors

load(filename : String, device : Device | Nil = nil) : self

new(c_module : TorchSys::CModule)

c_module

call_method(name : String, values : Array(IValue)) : IValue

each_named_parameter

Yields each named parameter (name, tensor). The tensors are owned by the wrapper and freed by GC as usual.

eval

finalize

forward(tensors : Array(Tensor)) : Tensor

Convenience path for tensor-only modules: raises unless the module returns a single tensor.

forward(values : Array(IValue)) : IValue

General path through IValues, for modules with non-tensor inputs/outputs.

forward(*tensors : Tensor) : Tensor

save(filename : String) : Nil

to(device : Device, kind : Kind | Nil = nil, non_blocking : Bool = false) : self

train