nn.types

interface Layer[T] {
	output_shape() []int
	variables() []&autograd.Variable[T]
	forward(input &autograd.Variable[T]) !&autograd.Variable[T]
}

Layer is a generic interface for a neural network layer.

interface Loss[T] {
	loss(input &autograd.Variable[T], target &vtl.Tensor[T]) !&autograd.Variable[T]
}

interface Optimizer[T] {
mut:
	params        []&autograd.Variable[T]
	learning_rate f64
	update() !
	build_params(layers Layer[T])
}

Optimizer is a generic interface for all optimizers.