Function getCartesianProductOfValues

tfjs-layers/src/utils/generic_utils.ts:541–563

(...arrayOfValues: PossibleValues)

 * @return The cartesian product.
 */
export function getCartesianProductOfValues(...arrayOfValues: PossibleValues):
    PossibleValues {
  assert(arrayOfValues.length > 0, 'arrayOfValues is empty');

  for (const values of arrayOfValues) {
    assert(Array.isArray(values), 'one of the values is not an array');
    assert(values.length > 0, 'one of the values is empty');
  }

  return arrayOfValues.reduce((products, values) => {
    if (products.length === 0) {
      return values.map(value => [value]);
    }

    return values
        .map(value => {
          return products.map((prevValue) => [...prevValue, value]);
        })
        .reduce((flattenedProduct, unflattenedProduct) => {
          return flattenedProduct.concat(unflattenedProduct);
        }, []);
  }, [] as PossibleValues);
}