Moduleonnxruntime-0.1.0.0GHC2024

OnnxRuntime.Shape

Tensor shapes in row-major order, and the dimensions a model declares.

Synopsis

newtypenewtype Shape

A tensor's dimensions, outermost first; the empty shape is a scalar.

Instances3Eq, Ord, Show

valueelementCount :: Shape -> Int

valuestrides :: Shape -> [Int]

Row-major strides: how far apart neighbours along each dimension lie.

valueflatIndex :: Shape -> [Int] -> Maybe Int

The flat offset of a multi-index, when it lies inside the shape.

valueunflatIndex :: Shape -> Int -> Maybe [Int]

The multi-index of a flat offset, when it lies inside the shape.

datadata Dim

A dimension a model declares: a fixed size, or one left free under a symbolic name (empty when unnamed).

Instances2Eq, Show

valueaccepts :: [Dim] -> Shape -> Bool

Whether declared dimensions take a shape: the same rank, and every fixed dimension equal.