Tinkex.Types.Datum (Tinkex v0.3.4)

Training example containing model input and loss function inputs.

Mirrors Python tinker.types.Datum.

Summary

Types

t()

Functions

new(attrs)

Create a new Datum with automatic tensor conversion.

Types

t()

@type t() :: %Tinkex.Types.Datum{
  loss_fn_inputs: %{required(String.t()) => Tinkex.Types.TensorData.t()},
  model_input: Tinkex.Types.ModelInput.t()
}

Functions

new(attrs)

@spec new(map()) :: t()

Create a new Datum with automatic tensor conversion.

Converts:

Nx.Tensor → TensorData
Plain lists → TensorData (with dtype inference)
TensorData → passthrough