Module `Space.Multi_discrete`

type element = (int32, Nx.int32_elt) Nx.t

Multiple discrete choices with independent cardinalities.

val create : int array -> element t

create nvec is a multi-discrete space where element i is in [0; nvec.(i) - 1].

Raises Invalid_argument if nvec is empty or any nvec.(i) <= 0.