The index of the embedder head (i.e. output tensor) this embedding comes from. This is useful for multi-head models.
@ffi.Int() external int head_index;