UNPKG

@stdlib/strided

Version:
155 lines (144 loc) 5.11 kB
/* * @license Apache-2.0 * * Copyright (c) 2020 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 { ArrayLike } from '@stdlib/types/array'; /** * Callback invoked for each indexed strided array element. * * @param value - strided array element * @returns result */ type Unary = ( value: any ) => any; /** * Interface describing `mskunary`. */ interface Routine { /** * Applies a unary callback to elements in a strided input array according to elements in a strided mask array and assigns results to elements in a strided output array. * * @param arrays - array-like object containing one input array, a mask array, and one output array * @param shape - array-like object containing a single element, the number of indexed elements * @param strides - array-like object containing the stride lengths for the strided arrays * @param fcn - unary callback * * @example * var Float64Array = require( '@stdlib/array/float64' ); * var Uint8Array = require( '@stdlib/array/uint8' ); * * function scale( x ) { * return x * 10.0; * } * * var x = new Float64Array( [ 1.0, 2.0, 3.0, 4.0, 5.0 ] ); * var m = new Uint8Array( [ 0, 0, 1, 0, 0 ] ); * var y = new Float64Array( x.length ); * * var shape = [ x.length ]; * var strides = [ 1, 1, 1 ]; * * mskunary( [ x, m, y ], shape, strides, scale ); * * console.log( y ); * // => <Float64Array>[ 10.0, 20.0, 0.0, 40.0, 50.0 ] */ ( arrays: ArrayLike<ArrayLike<any>>, shape: ArrayLike<number>, strides: ArrayLike<number>, fcn: Unary ): void; /** * Applies a unary callback to elements in a strided input array according to elements in a strided mask array and assigns results to elements in a strided output array using alternative indexing semantics. * * @param arrays - array-like object containing one input array, a mask array, and one output array * @param shape - array-like object containing a single element, the number of indexed elements * @param strides - array-like object containing the stride lengths for strided arrays * @param offsets - array-like object containing the starting indices (i.e., index offsets) for the strided arrays * @param fcn - unary callback * * @example * var Float64Array = require( '@stdlib/array/float64' ); * var Uint8Array = require( '@stdlib/array/uint8' ); * * function scale( x ) { * return x * 10.0; * } * * var x = new Float64Array( [ 1.0, 2.0, 3.0, 4.0, 5.0 ] ); * var m = new Uint8Array( [ 0, 0, 1, 0, 0 ] ); * var y = new Float64Array( x.length ); * * var shape = [ x.length ]; * var strides = [ 1, 1, 1 ]; * var offsets = [ 0, 0, 0 ]; * * mskunary.ndarray( [ x, m, y ], shape, strides, offsets, scale ); * * console.log( y ); * // => <Float64Array>[ 10.0, 20.0, 0.0, 40.0, 50.0 ] */ ndarray( arrays: ArrayLike<ArrayLike<any>>, shape: ArrayLike<number>, strides: ArrayLike<number>, offsets: ArrayLike<number>, fcn: Unary ): void; } /** * Applies a unary callback to elements in a strided input array according to elements in a strided mask array and assigns results to elements in a strided output array. * * @param arrays - array-like object containing one input array, a mask array, and one output array * @param shape - array-like object containing a single element, the number of indexed elements * @param strides - array-like object containing the stride lengths for the strided arrays * @param fcn - unary callback * * @example * var Float64Array = require( '@stdlib/array/float64' ); * var Uint8Array = require( '@stdlib/array/uint8' ); * * function scale( x ) { * return x * 10.0; * } * * var x = new Float64Array( [ 1.0, 2.0, 3.0, 4.0, 5.0 ] ); * var m = new Uint8Array( [ 0, 0, 1, 0, 0 ] ); * var y = new Float64Array( x.length ); * * var shape = [ x.length ]; * var strides = [ 1, 1, 1 ]; * * mskunary( [ x, m, y ], shape, strides, scale ); * * console.log( y ); * // => <Float64Array>[ 10.0, 20.0, 0.0, 40.0, 50.0 ] * * @example * var Float64Array = require( '@stdlib/array/float64' ); * var Uint8Array = require( '@stdlib/array/uint8' ); * * function scale( x ) { * return x * 10.0; * } * * var x = new Float64Array( [ 1.0, 2.0, 3.0, 4.0, 5.0 ] ); * var m = new Uint8Array( [ 0, 0, 1, 0, 0 ] ); * var y = new Float64Array( x.length ); * * var shape = [ x.length ]; * var strides = [ 1, 1, 1 ]; * var offsets = [ 0, 0, 0 ]; * * mskunary.ndarray( [ x, m, y ], shape, strides, offsets, scale ); * * console.log( y ); * // => <Float64Array>[ 10.0, 20.0, 0.0, 40.0, 50.0 ] */ declare var mskunary: Routine; // EXPORTS // export = mskunary;