Settings View Source Nx.Defn.Token (Nx v0.4.1)

A defn token used by hooks.

documentation-for-compilers

Documentation for compilers

The token has a hooks field as a list of maps of the shape:

%{
  expr: Nx.Tensor.t | Nx.Container.t,
  name: atom(),
  callback: (Nx.Tensor.t | Nx.Container.t -> term()) | nil
}

The hooks field must only be accessed by defn compilers.