mskunary3d

Apply a unary callback to elements in a three-dimensional nested input array according to elements in a three-dimensional nested mask array and assign results to elements in a three-dimensional nested output array.

Usage

var mskunary3d = require( '@stdlib/array/base/mskunary3d' );

mskunary3d( arrays, shape, fcn )

Applies a unary callback to elements in a three-dimensional nested input array according to elements in a three-dimensional nested mask array and assigns results to elements in a three-dimensional nested output array.

var abs = require( '@stdlib/math/base/special/abs' );

var x = [ [ [ -1.0, -2.0 ], [ -3.0, -4.0 ] ] ];
var mask = [ [ [ 0, 1 ], [ 0, 0 ] ] ];

var shape = [ 1, 2, 2 ];

// Compute the absolute values in-place:
mskunary3d( [ x, mask, x ], shape, abs );
// x => [ [ [ 1.0, -2.0 ], [ 3.0, 4.0 ] ] ]

The function accepts the following arguments:

arrays: array-like object containing one input nested array, an input nested mask array, and one output nested array.
shape: array shape.
fcn: unary function to apply.

Notes

The function assumes that the input and output arrays have the same shape.
An element in an input array is "masked" if the corresponding element in the mask array is non-zero.

Examples

var discreteUniform = require( '@stdlib/random/base/discrete-uniform' ).factory;
var bernoulli = require( '@stdlib/random/base/bernoulli' ).factory;
var filled3dBy = require( '@stdlib/array/base/filled3d-by' );
var zeros3d = require( '@stdlib/array/base/zeros3d' );
var abs = require( '@stdlib/math/base/special/abs' );
var mskunary3d = require( '@stdlib/array/base/mskunary3d' );

var shape = [ 3, 3, 3 ];

var x = filled3dBy( shape, discreteUniform( -100, 100 ) );
console.log( x );

var mask = filled3dBy( shape, bernoulli( 0.5 ) );
console.log( mask );

var y = zeros3d( shape );
console.log( y );

mskunary3d( [ x, mask, y ], shape, abs );
console.log( y );