Function mergeElementShape

tfjs-converter/src/executor/tensor_utils.ts:87–113

(
    elementShapeA: number|number[], elementShapeB: number|number[])

}

export function mergeElementShape(
    elementShapeA: number|number[], elementShapeB: number|number[]): number|
    number[] {
  if (typeof elementShapeA === 'number') {
    return elementShapeB;
  }
  if (typeof elementShapeB === 'number') {
    return elementShapeA;
  }

  if (elementShapeA.length !== elementShapeB.length) {
    throw new Error(`Incompatible ranks during merge: ${elementShapeA} vs. ${
        elementShapeB}`);
  }

  const result: number[] = [];
  for (let i = 0; i < elementShapeA.length; ++i) {
    const dim0 = elementShapeA[i];
    const dim1 = elementShapeB[i];
    if (dim0 >= 0 && dim1 >= 0 && dim0 !== dim1) {
      throw new Error(`Incompatible shape during merge: ${elementShapeA} vs. ${
          elementShapeB}`);
    }
    result[i] = dim0 >= 0 ? dim0 : dim1;
  }
  return result;
}