Tinkex.Types.TensorData (Tinkex v0.1.17)

Numerical tensor data for training.

Mirrors Python tinker.types.TensorData.

IMPORTANT: Only int64 and float32 dtypes are supported by the backend. This module performs aggressive casting to match Python SDK behavior.

Summary

t()

from_nx(tensor)

Create TensorData from an Nx tensor.

Convert TensorData back to an Nx tensor.

@type t() :: %Tinkex.Types.TensorData{
  data: [number()],
  dtype: Tinkex.Types.TensorDtype.t(),
  shape: [non_neg_integer()] | nil
}

@spec from_nx(Nx.Tensor.t()) :: t()

Create TensorData from an Nx tensor.

Performs aggressive type casting to match Python SDK:

@spec to_nx(t()) :: Nx.Tensor.t()

Convert TensorData back to an Nx tensor.