GatherAxes

import { GatherAxes } from "destack:tensor";

Axis mapping for tensor gather.

export struct GatherAxes<const OffsetCount: usize, const CollapsedCount: usize, const StartIndexCount: usize> {
    offset: [Axis; OffsetCount];
    collapsedSlice: [Axis; CollapsedCount];
    startIndexMap: [Axis; StartIndexCount];
    indexVector: Axis;
}
language/library/src/tensor/axes.ds:51:67

Members

offset: [Axis; OffsetCount]

Offset axes in the output tensor.

collapsedSlice: [Axis; CollapsedCount]

Collapsed slice axes in the operand tensor.

startIndexMap: [Axis; StartIndexCount]

Mapping from index components to operand axes.

indexVector: Axis

Axis containing index vectors in the indices tensor.