UNPKG

@stdlib/array

Version:
70 lines (61 loc) 2.12 kB
/* * @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 /// <reference types="@stdlib/types"/> import { Array5D } from '@stdlib/types/array'; import { Shape5D } from '@stdlib/types/ndarray'; /** * Ternary callback. * * @param v1 - element from first input array * @param v2 - element from second input array * @param v3 - element from third input array * @returns result */ type Ternary<T, U, V, W> = ( v1: T, v2: U, v3: V ) => W; /** * Applies a ternary callback to elements in three five-dimensional nested input arrays and assigns results to elements in a five-dimensional nested output array. * * ## Notes * * - The function assumes that the input and output arrays have the same shape. * * @param arrays - array containing three input nested arrays and one output nested array * @param shape - array shape * @param fcn - ternary callback * * @example * var add = require( '@stdlib/math/base/ops/add3' ); * var ones5d = require( '@stdlib/array/base/ones5d' ); * var zeros5d = require( '@stdlib/array/base/zeros5d' ); * * var shape = [ 1, 1, 1, 2, 2 ]; * * var x = ones5d( shape ); * var y = ones5d( shape ); * var z = ones5d( shape ); * var out = zeros5d( shape ); * * ternary5d( [ x, y, z, out ], shape, add ); * * console.log( out ); * // => [ [ [ [ [ 3.0, 3.0 ], [ 3.0, 3.0 ] ] ] ] ] */ declare function ternary5d<T = unknown, U = unknown, V = unknown, W = unknown>( arrays: [ Array5D<T>, Array5D<U>, Array5D<V>, Array5D<W> ], shape: Shape5D, fcn: Ternary<T, U, V, W> ): void; // EXPORTS // export = ternary5d;