tech.compute.tensor.dimensions.select — techascent/tech.compute 3.2

Selecting subsets from a larger set of dimensions leads to its own algebra.

apply-select-arg-to-dimension clj

(apply-select-arg-to-dimension dim select-arg)

Given a dimension and select argument, create a new dimension with the selection applied.

dimensions->simpified-dimensions clj

(dimensions->simpified-dimensions dimension-seq stride-seq)

Given the dimensions post selection, produce a new dimension sequence combined with an offset that lets us know how much we should offset the base storage type. Simplification is important because it allows a backend to hit more fast paths. Returns: {:dimension-seq dimension-seq :offset offset}