docs menu

Module Nx.Mesh

Named axes over a set's devices.

#type +'d t

The type for named axes over a set of brand 'd.

#val v : 'd Devices.t -> (string * int) list -> 'd t

v s axes lays s's devices in row-major order over the named axes.

Raises Invalid_argument unless the extents are positive and multiply to the number of s's devices and the names are distinct.