UNPKG

@stdlib/array

Version:
70 lines (61 loc) 2.1 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 { Array3D } from '@stdlib/types/array'; import { Shape3D } 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 three-dimensional nested input arrays and assigns results to elements in a three-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 ones3d = require( '@stdlib/array/base/ones3d' ); * var zeros3d = require( '@stdlib/array/base/zeros3d' ); * * var shape = [ 1, 2, 2 ]; * * var x = ones3d( shape ); * var y = ones3d( shape ); * var z = ones3d( shape ); * var out = zeros3d( shape ); * * ternary3d( [ x, y, z, out ], shape, add ); * * console.log( out ); * // => [ [ [ 3.0, 3.0 ], [ 3.0, 3.0 ] ] ] */ declare function ternary3d<T = unknown, U = unknown, V = unknown, W = unknown>( arrays: [ Array3D<T>, Array3D<U>, Array3D<V>, Array3D<W> ], shape: Shape3D, fcn: Ternary<T, U, V, W> ): void; // EXPORTS // export = ternary3d;