/* * @license Apache-2.0 * * Copyright (c) 2023 The Stdlib Authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ // TypeScript Version: 4.1 /// import { ArrayLike } from '@stdlib/types/array'; import { typedndarray } from '@stdlib/types/ndarray'; /** * Broadcasts ndarrays to a common shape. * * ## Notes * * - The function throws an error if a provided broadcast-incompatible ndarrays. * - If a provided ndarray has a shape matching the common shape, the function returns the provided ndarray. * - If a provided ndarray has a different (broadcast compatible) shape than the common shape, the function returns a new **read-only** ndarray view of the provided ndarray's data. The view is typically **not** contiguous. As more than one element of a returned view may refer to the same memory location, writing to a view may affect multiple elements. If you need to write to an input array, copy the array before broadcasting. * * @param arrays - input arrays * @throws input arrays must be broadcast compatible * @returns list of broadcasted arrays * * @example * var array = require( '@stdlib/ndarray-array' ); * var zeros = require( '@stdlib/ndarray-zeros' ); * * var x = array( [ [ 1, 2 ], [ 3, 4 ] ] ); * // returns * * var shx = x.shape; * // returns [ 2, 2 ] * * var y1 = zeros( [ 3, 2, 2 ] ); * // returns * * var shy = y1.shape; * // returns [ 3, 2, 2 ] * * var out = maybeBroadcastArrays( [ x, y ] ); * // returns * * var x2 = out[ 0 ]; * // returns * * var y2 = out[ 1 ]; * // returns * * shx = x2.shape; * // returns [ 3, 2, 2 ] * * shy = y2.shape; * // returns [ 3, 2, 2 ] * * var v = x2.get( 0, 0, 0 ); * // returns 1 * * v = x2.get( 0, 0, 1 ); * // returns 2 * * v = x2.get( 1, 0, 0 ); * // returns 1 * * v = x2.get( 1, 1, 0 ); * // returns 3 * * v = x2.get( 2, 0, 0 ); * // returns 1 * * v = x2.get( 2, 1, 1 ); * // returns 4 */ declare function maybeBroadcastArrays( arrays: ArrayLike> ): Array>; /** * Broadcasts ndarrays to a common shape. * * ## Notes * * - The function throws an error if a provided broadcast-incompatible ndarrays. * - If a provided ndarray has a shape matching the common shape, the function returns the provided ndarray. * - If a provided ndarray has a different (broadcast compatible) shape than the common shape, the function returns a new **read-only** ndarray view of the provided ndarray's data. The view is typically **not** contiguous. As more than one element of a returned view may refer to the same memory location, writing to a view may affect multiple elements. If you need to write to an input array, copy the array before broadcasting. * * @param arrays - input arrays * @throws input arrays must be broadcast compatible * @returns list of broadcasted arrays * * @example * var array = require( '@stdlib/ndarray-array' ); * var zeros = require( '@stdlib/ndarray-zeros' ); * * var x = array( [ [ 1, 2 ], [ 3, 4 ] ] ); * // returns * * var shx = x.shape; * // returns [ 2, 2 ] * * var y1 = zeros( [ 3, 2, 2 ] ); * // returns * * var shy = y1.shape; * // returns [ 3, 2, 2 ] * * var out = maybeBroadcastArrays( x, y ); * // returns * * var x2 = out[ 0 ]; * // returns * * var y2 = out[ 1 ]; * // returns * * shx = x2.shape; * // returns [ 3, 2, 2 ] * * shy = y2.shape; * // returns [ 3, 2, 2 ] * * var v = x2.get( 0, 0, 0 ); * // returns 1 * * v = x2.get( 0, 0, 1 ); * // returns 2 * * v = x2.get( 1, 0, 0 ); * // returns 1 * * v = x2.get( 1, 1, 0 ); * // returns 3 * * v = x2.get( 2, 0, 0 ); * // returns 1 * * v = x2.get( 2, 1, 1 ); * // returns 4 */ declare function maybeBroadcastArrays( ...arrays: Array> ): Array>; // EXPORTS // export = maybeBroadcastArrays;