Function numElements

tensorboard/components/tensor_widget/shape-utils.ts:23–29

(shape: Shape)

/** Compute total element count based on shape. */
export function numElements(shape: Shape): number {
  let output = 1;
  shape.forEach((dimSize) => {
    output *= dimSize;
  });
  return output;
}

/**
 * Format tensor shape as a string for display.