DotAxes

import { DotAxes } from "destack:tensor";

Axis mapping for tensor dot.

export struct DotAxes<const BatchCount: usize, const ContractCount: usize> {
    leftBatch: [Axis; BatchCount];
    rightBatch: [Axis; BatchCount];
    leftContract: [Axis; ContractCount];
    rightContract: [Axis; ContractCount];
}
language/library/src/tensor/axes.ds:6:18

Members

leftBatch: [Axis; BatchCount]

Batch axes in the left tensor.

rightBatch: [Axis; BatchCount]

Batch axes in the right tensor.

leftContract: [Axis; ContractCount]

Contracted axes in the left tensor.

rightContract: [Axis; ContractCount]

Contracted axes in the right tensor.