jax-js
    Preparing search index...

    Type Alias DotDimensionNumbers

    Dimension numbers for general dot() primitive.

    Contracting dimensions act as a tensor contraction (reduction) along the given axis. They must be the same size in both operands. Batch dimensions are treated as vectorized, leading batch dimensions.

    The return value has a shape where the first dimensions are shared batch dimensions, followed by lhs non-contracting dimensions, followed by rhs non-contracting dimensions.

    type DotDimensionNumbers = {
        lhsBatchDims?: number[];
        lhsContractingDims?: number[];
        rhsBatchDims?: number[];
        rhsContractingDims?: number[];
    }

    Properties

    lhsBatchDims?: number[]
    lhsContractingDims?: number[]
    rhsBatchDims?: number[]
    rhsContractingDims?: number[]
    Properties