Interface INTMController<T>

Namespace

AiDotNet.MetaLearning.Algorithms

Assembly

AiDotNet.dll

Interface for NTM controller.

public interface INTMController<T>

Type Parameters

T

The numeric type.

Methods

Forward(Tensor<T>, List<Tensor<T>>)

Forward pass through the controller.

Tensor<T> Forward(Tensor<T> input, List<Tensor<T>> readContents)

Parameters

input Tensor<T>

The input tensor.

readContents List<Tensor<T>>

The previous read contents.

Returns

Tensor<T>

The controller output.

GenerateAddVector(Tensor<T>)

Generates add vector for writing.

Tensor<T> GenerateAddVector(Tensor<T> output)

Parameters

output Tensor<T>

The controller output.

Returns

Tensor<T>

The add vector.

GenerateEraseVector(Tensor<T>)

Generates erase vector for writing.

Tensor<T> GenerateEraseVector(Tensor<T> output)

Parameters

output Tensor<T>

The controller output.

Returns

Tensor<T>

The erase vector.

GenerateOutput(Tensor<T>, List<Tensor<T>>)

Generates final output.

Tensor<T> GenerateOutput(Tensor<T> output, List<Tensor<T>> readContents)

Parameters

output Tensor<T>

The controller output.

readContents List<Tensor<T>>

The current read contents.

Returns

Tensor<T>

The final output.

GenerateReadKeys(Tensor<T>)

Generates read keys for all read heads.

List<Tensor<T>> GenerateReadKeys(Tensor<T> output)

Parameters

output Tensor<T>

The controller output.

Returns

List<Tensor<T>>

List of read keys.

GenerateWriteKey(Tensor<T>)

Generates write key.

Tensor<T> GenerateWriteKey(Tensor<T> output)

Parameters

output Tensor<T>

The controller output.

Returns

Tensor<T>

The write key.

GetParameters()

Gets controller parameters.

Vector<T> GetParameters()

Returns

Vector<T>

The parameter vector.

Reset()

Resets controller state.

void Reset()

SetParameters(Vector<T>)

Sets controller parameters (updates internal weights).

void SetParameters(Vector<T> parameters)

Parameters

parameters Vector<T>

The parameter vector to set.