feat: implement broadcastSizes and broadcastTo functions for matrix operations

Author: dvd101x

src/function/matrix/broadcastSizes.js

@@ -0,0 +1,36 @@
+import { broadcastSizes } from '../../utils/array.js'
+import { factory } from '../../utils/factory.js'
+import { isMatrix } from '../../utils/is.js'
+
+const name = 'broadcastSizes'
+const dependencies = ['typed']
+
+export const createBroadcastSizes = /* #__PURE__ */ factory(name, dependencies, ({ typed }) => {
+  /**
+   * Calculate the broadcasted size of one or more matrices or arrays.
+   * Always returns an Array containing numbers.
+   *
+   * Syntax:
+   *
+   *     math.broadcastSizes(x, y)
+   *     math.broadcastSizes(x, y, ...)
+   *
+   * Examples:
+   *
+   *     math.broadcastSizes([2, 3])          // returns [2, 3]
+   *     math.broadcastSizes([2, 3], [3])     // returns [2, 3]
+   *     math.broadcastSizes([1, 2, 3], [[1, 2, 3]])  // returns [[1, 2, 3]]
+   *
+   * See also:
+   *
+   *     size, reshape, squeeze, broadcastTo
+   *
+   * @param {...(Array|Matrix)} x  One or more matrices or arrays
+   * @return {Array} A vector with the broadcasted size.
+   */
+  return typed(name, {
+    '...Array': broadcastSizes,
+    '...Matrix': matrices => broadcastSizes(...matrices.map(m => m.toArray())),
+    '...Array|Matrix': collections => broadcastSizes(...collections.map(c => isMatrix(c) ? c.toArray() : c))
+  })
+})

src/function/matrix/broadcastTo.js

@@ -0,0 +1,34 @@
+import { broadcastTo } from '../../utils/array.js'
+import { factory } from '../../utils/factory.js'
+
+const name = 'broadcastTo'
+const dependencies = ['typed']
+
+export const createBroadcastTo = /* #__PURE__ */ factory(name, dependencies, ({ typed }) => {
+    /**
+     * Broadcast an array to a specified size.
+     *
+     * Syntax:
+     *
+     *     math.broadcastTo(x, size)
+     *
+     * Examples:
+     *
+     *     math.broadcastTo([1, 2, 3], [2, 3])          // returns [[1, 2, 3], [1, 2, 3]]
+     *     math.broadcastTo([2, 3], [2, 2])             // returns [[2, 3], [2, 3]]
+     *
+     * See also:
+     *
+     *     size, reshape, squeeze, broadcastSizes
+     *
+     * @param {Array|Matrix} x  The array or matrix to broadcast
+     * @param {Array|Matrix} size  The target size
+     * @return {Array} The broadcasted array
+     */
+  return typed(name, {
+    'Array, Array': broadcastTo,
+    'Array, Matrix': (arr, size) => broadcastTo(arr, size.toArray()),
+    'Matrix, Array': (M, size) => M.create(broadcastTo(M.toArray(), size)),
+    'Matrix, Matrix': (M1, size) => M1.create(broadcastTo(M1.toArray(), size.toArray()))
+  })
+})