Function getEinsumComputePath

  • getEinsumComputePath(summedDims, idDims): {
        path: number[];
        steps: number[][];
    }

  • Gets path of computation for einsum.

    Parameters

    • summedDims: number[]

      indices to the dimensions being summed over.

    • idDims: number[][]

      A look up table for the dimensions present in each input tensor. Each consituent array contains indices for the dimensions in the corresponding input tensor.

    Returns {
        path: number[];
        steps: number[][];
    }

    A map with two fields:

    • path: The path of computation, with each element indicating the dimension being summed over after the element-wise multiplication in that step.
    • steps: With the same length as path. Each element contains the indices to the input tensors being used for element-wise multiplication in the corresponding step.
    • path: number[]
    • steps: number[][]

Settings

Member Visibility

Theme

Generated using TypeDoc