Broadcasting functions

src/expression/embeddedDocs/embeddedDocs.js

@@ -143,6 +143,9 @@ import { partitionSelectDocs } from './function/matrix/partitionSelect.js'
 import { rangeDocs } from './function/matrix/range.js'
 import { reshapeDocs } from './function/matrix/reshape.js'
 import { resizeDocs } from './function/matrix/resize.js'
+import { broadcastMatricesDocs } from './function/matrix/broadcastMatrices.js'
+import { broadcastToDocs } from './function/matrix/bradcastTo.js'
+import { broadcastSizesDocs } from './function/matrix/broadcastSizes.js'
 import { rotateDocs } from './function/matrix/rotate.js'
 import { rotationMatrixDocs } from './function/matrix/rotationMatrix.js'
 import { rowDocs } from './function/matrix/row.js'
@@ -481,6 +484,9 @@ export const embeddedDocs = {
   partitionSelect: partitionSelectDocs,
   range: rangeDocs,
   resize: resizeDocs,
+  broadcastMatrices: broadcastMatricesDocs,
+  broadcastTo: broadcastToDocs,
+  broadcastSizes: broadcastSizesDocs,
   reshape: reshapeDocs,
   rotate: rotateDocs,
   rotationMatrix: rotationMatrixDocs,

src/expression/embeddedDocs/function/matrix/bradcastTo.js

@@ -0,0 +1,15 @@
+export const broadcastToDocs = {
+  name: 'broadcastTo',
+  category: 'Matrix',
+  syntax: [
+    'broadcastTo(A, size)'
+  ],
+  description: 'Broadcast a matrix to a compatible size',
+  examples: [
+    'broadcastTo([1, 2, 3], [3, 3])',
+    'broadcastTo([1, 2; 3, 4], [2, 2])'
+  ],
+  seealso: [
+    'size', 'reshape', 'broadcastSizes', 'broadcastMatrices'
+  ]
+}

src/expression/embeddedDocs/function/matrix/broadcastMatrices.js

@@ -0,0 +1,15 @@
+export const broadcastMatricesDocs = {
+  name: 'broadcastMatrices',
+  category: 'Matrix',
+  syntax: [
+    'broadcastMatrices(A, B)'
+  ],
+  description: 'Broadcast two matrices to compatible sizes',
+  examples: [
+    'broadcastMatrices([1, 2, 3], [[1], [2], [3]])',
+    'broadcastMatrices([1, 2; 3, 4], [5, 6; 7, 8])'
+  ],
+  seealso: [
+    'size', 'reshape', 'broadcastSizes', 'broadcastTo'
+  ]
+}

src/expression/embeddedDocs/function/matrix/broadcastSizes.js

@@ -0,0 +1,15 @@
+export const broadcastSizesDocs = {
+  name: 'broadcastSizes',
+  category: 'Matrix',
+  syntax: [
+    'broadcastSizes(sizeA, sizeB)'
+  ],
+  description: 'Broadcast the sizes of matrices to a compatible size',
+  examples: [
+    'broadcastSizes([3, 1, 3], [3, 3])',
+    'broadcastSizes([2, 1], [2, 2])'
+  ],
+  seealso: [
+    'size', 'reshape', 'broadcastTo', 'broadcastMatrices'
+  ]
+}

src/factoriesAny.js

@@ -88,6 +88,9 @@ export { createOnes } from './function/matrix/ones.js'
 export { createRange } from './function/matrix/range.js'
 export { createReshape } from './function/matrix/reshape.js'
 export { createResize } from './function/matrix/resize.js'
+export { createBroadcastMatrices } from './function/matrix/broadcastMatrices.js'
+export { createBroadcastSizes } from './function/matrix/broadcastSizes.js'
+export { createBroadcastTo } from './function/matrix/broadcastTo.js'
 export { createRotate } from './function/matrix/rotate.js'
 export { createRotationMatrix } from './function/matrix/rotationMatrix.js'
 export { createRow } from './function/matrix/row.js'

src/function/matrix/broadcastMatrices.js

@@ -0,0 +1,46 @@
+import { broadcastArrays } from '../../utils/array.js'
+import { factory } from '../../utils/factory.js'
+import { isMatrix } from '../../utils/is.js'
+
+const name = 'broadcastMatrices'
+const dependencies = ['typed']
+
+export const createBroadcastMatrices = /* #__PURE__ */ factory(name, dependencies, ({ typed }) => {
+  /**
+   * Broadcast multiple matrices together.
+   * Return and array of matrices with the broadcasted sizes.
+   *
+   * Syntax:
+   *
+   *     math.broadcastMatrices(x, y)
+   *     math.broadcastMatrices(x, y, ...)
+   *
+   * Examples:
+   *
+   *     math.broadcastMatrices([1, 2], [[3], [4]])   // returns [[[1, 2], [1, 2]], [[3, 3], [4, 4]]]
+   *     math.broadcastMatrices([2, 3])               // returns [[2, 3]]
+   *     math.broadcastMatrices([2, 3], [3, 1])       // returns [[2, 3], [3, 1]]
+   *
+   * See also:
+   *
+   *     size, reshape, broadcastSizes, broadcastTo
+   *
+   * History:
+   *
+   *     v15.1.1 created
+   * @param {...(Array|Matrix)} x  One or more matrices or arrays
+   * @return {Array[Array|Matrix]} An array of matrices with the broadcasted sizes.
+   */
+  return typed(name, {
+    '...Array|Matrix': collections => {
+      const areMatrices = collections.map(isMatrix)
+      if (areMatrices.includes(true)) {
+        const arrays = collections.map((c, i) => areMatrices[i] ? c.valueOf() : c)
+        const broadcastedArrays = broadcastArrays(...arrays)
+        const broadcastedCollections = broadcastedArrays.map((arr, i) => areMatrices[i] ? collections[i].create(arr) : arr)
+        return broadcastedCollections
+      }
+      return broadcastArrays(...collections)
+    }
+  })
+})

src/function/matrix/broadcastSizes.js

@@ -0,0 +1,44 @@
+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, 1])  // returns [1, 2, 3]
+   *
+   * See also:
+   *
+   *     size, reshape, squeeze, broadcastTo
+   *
+   * History:
+   *
+   *     v15.1.1 created
+   * @param {...(Array|Matrix)} x  One or more matrices or arrays
+   * @return {Array} A vector with the broadcasted size.
+   */
+  return typed(name, {
+    '...Array|Matrix': collections => {
+      const areMatrices = collections.map(isMatrix)
+      if (areMatrices.includes(true)) {
+        const arrays = collections.map((c, i) => areMatrices[i] ? c.valueOf() : c)
+        return broadcastSizes(...arrays)
+      }
+      return broadcastSizes(...collections)
+    }
+  })
+})

src/function/matrix/broadcastTo.js

@@ -0,0 +1,38 @@
+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
+     *
+     * History:
+     *
+     *     v15.1.1 created
+     *
+     * @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()))
+  })
+})

src/utils/array.js

@@ -810,7 +810,7 @@ export function broadcastArrays (...arrays) {
     throw new Error('Insufficient number of arguments in function broadcastArrays')
   }
   if (arrays.length === 1) {
-    return arrays[0]
+    return arrays
   }
   const sizes = arrays.map(function (array) { return arraySize(array) })
   const broadcastedSize = broadcastSizes(...sizes)

test/unit-tests/function/matrix/broadcastMatrices.test.js

@@ -0,0 +1,35 @@
+import assert from 'assert'
+import math from '../../../../src/defaultInstance.js'
+
+describe('broadcastMatrices', function () {
+  const matrix = math.matrix
+  const broadcastMatrices = math.broadcastMatrices
+  const A = [[1], [2], [3]]
+  const B = [[10, 20, 30]]
+  const C = [100]
+  const broadcastedA = [[1, 1, 1], [2, 2, 2], [3, 3, 3]]
+  const broadcastedB = [[10, 20, 30], [10, 20, 30], [10, 20, 30]]
+  const broadcastedC = [[100, 100, 100], [100, 100, 100], [100, 100, 100]]
+
+  it('should broadcast matrices', function () {
+    assert.deepStrictEqual(broadcastMatrices(matrix(A), matrix(B)), [matrix(broadcastedA), matrix(broadcastedB)])
+    assert.deepStrictEqual(broadcastMatrices(matrix(A), matrix(C)), [matrix(A), matrix([[100], [100], [100]])])
+    assert.deepStrictEqual(broadcastMatrices(matrix(B), matrix(A)), [matrix(broadcastedB), matrix(broadcastedA)])
+    assert.deepStrictEqual(broadcastMatrices(matrix(A), matrix(B), matrix(C)), [matrix(broadcastedA), matrix(broadcastedB), matrix(broadcastedC)])
+  })
+
+  it('should broadcast arrays', function () {
+    assert.deepStrictEqual(broadcastMatrices(A, B), [broadcastedA, broadcastedB])
+    assert.deepStrictEqual(broadcastMatrices(B, A), [broadcastedB, broadcastedA])
+    assert.deepStrictEqual(broadcastMatrices(A, B, C), [broadcastedA, broadcastedB, broadcastedC])
+    assert.deepStrictEqual(broadcastMatrices(A), [A])
+    assert.deepStrictEqual(broadcastMatrices(B, C, A), [broadcastedB, broadcastedC, broadcastedA])
+  })
+
+  it('should throw an error if sizes are not compatible', function () {
+    assert.throws(function () { broadcastMatrices(matrix([[1, 2], [3, 4]]), matrix([[1, 2, 3]])) }, /Error: shape mismatch: /)
+    assert.throws(function () { broadcastMatrices([[1, 2], [3, 4]], matrix([[1, 2, 3]])) }, /Error: shape mismatch: /)
+    assert.throws(function () { broadcastMatrices(matrix([[1, 2], [3, 4]]), [[1, 2, 3]]) }, /Error: shape mismatch: /)
+    assert.throws(function () { broadcastMatrices([[1, 2], [3, 4]], [[1, 2, 3]]) }, /Error: shape mismatch: /)
+  })
+})

test/unit-tests/function/matrix/broadcastSizes.test.js

@@ -0,0 +1,27 @@
+import assert from 'assert'
+import math from '../../../../src/defaultInstance.js'
+
+describe('broadcastSizes', function () {
+  const broadcastSizes = math.broadcastSizes
+  const matrix = math.matrix
+
+  it('should broadcast sizes', function () {
+    assert.deepStrictEqual(broadcastSizes([2, 3]), [2, 3])
+    assert.deepStrictEqual(broadcastSizes([3, 3], [3, 1]), [3, 3])
+    assert.deepStrictEqual(broadcastSizes([2, 1], [1, 3]), [2, 3])
+    assert.deepStrictEqual(broadcastSizes([5, 4, 3], [1, 4, 1]), [5, 4, 3])
+    assert.deepStrictEqual(broadcastSizes([3], [2, 3]), [2, 3])
+    assert.deepStrictEqual(broadcastSizes([1, 3], [2, 1]), [2, 3])
+  })
+
+  it('should throw an error if sizes are not compatible', function () {
+    assert.throws(function () { broadcastSizes([2, 3], [3, 2]) }, /Error: shape mismatch: /)
+    assert.throws(function () { broadcastSizes([2, 3], [2, 3, 4]) }, /Error: shape mismatch: /)
+  })
+
+  it('should broadcast sizes of mixed arrays and matrices', function () {
+    assert.deepStrictEqual(broadcastSizes([3, 3], matrix([3, 1])), [3, 3])
+    assert.deepStrictEqual(broadcastSizes(matrix([2, 1]), [1, 3]), [2, 3])
+    assert.deepStrictEqual(broadcastSizes(matrix([5, 4, 3]), matrix([1, 4, 1])), [5, 4, 3])
+  })
+})

test/unit-tests/function/matrix/broadcastTo.test.js

@@ -0,0 +1,29 @@
+import assert from 'assert'
+import math from '../../../../src/defaultInstance.js'
+
+describe('broadcastTo', function () {
+  const broadcastTo = math.broadcastTo
+  const matrix = math.matrix
+
+  it('should broadcast arrays to a given size', function () {
+    assert.deepStrictEqual(broadcastTo([1, 2, 3], [2, 3]), [[1, 2, 3], [1, 2, 3]])
+    assert.deepStrictEqual(broadcastTo([2, 3], [2, 2]), [[2, 3], [2, 3]])
+  })
+
+  it('should broadcast matrices to a given size', function () {
+    assert.deepStrictEqual(broadcastTo(matrix([1, 2, 3]), [2, 3]), matrix([[1, 2, 3], [1, 2, 3]]))
+    assert.deepStrictEqual(broadcastTo(matrix([2, 3]), [2, 2]), matrix([[2, 3], [2, 3]]))
+  })
+
+  it('should broadcast mixed arrays and matrices to a given size', function () {
+    assert.deepStrictEqual(broadcastTo([1, 2, 3], matrix([2, 3])), [[1, 2, 3], [1, 2, 3]])
+    assert.deepStrictEqual(broadcastTo(matrix([2, 3]), [2, 2]), matrix([[2, 3], [2, 3]]))
+    assert.deepStrictEqual(broadcastTo(matrix([1, 2, 3]), matrix([2, 3])), matrix([[1, 2, 3], [1, 2, 3]]))
+    assert.deepStrictEqual(broadcastTo([2, 3], matrix([2, 2])), [[2, 3], [2, 3]])
+  })
+
+  it('should throw an error if sizes are not compatible', function () {
+    assert.throws(function () { broadcastTo([1, 2], [2, 3]) }, /Error: shape mismatch: /)
+    assert.throws(function () { broadcastTo(matrix([1, 2]), [2, 3]) }, /Error: shape mismatch: /)
+  })
+})

test/unit-tests/utils/array.test.js

@@ -702,9 +702,9 @@ describe('util.array', function () {
     })
 
     it('should broadcast leave arrays as such when only one is supplied', function () {
-      assert.deepStrictEqual(broadcastArrays([1, 2]), [1, 2], [3, 4])
-      assert.deepStrictEqual(broadcastArrays([[3], [4]]), [[3], [4]])
-      assert.deepStrictEqual(broadcastArrays([[5, 6]]), [[5, 6]])
+      assert.deepStrictEqual(broadcastArrays([1, 2])[0], [1, 2])
+      assert.deepStrictEqual(broadcastArrays([[3], [4]])[0], [[3], [4]])
+      assert.deepStrictEqual(broadcastArrays([[5, 6]])[0], [[5, 6]])
     })
 
     it('should throw an arryor when the broadcasting rules don\'t apply', function () {