-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfactory.js
118 lines (108 loc) · 3.18 KB
/
factory.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/**
* @license Apache-2.0
*
* Copyright (c) 2025 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.
*/
'use strict';
// MODULES //
var isFunction = require( '@stdlib/assert-is-function' );
var format = require( '@stdlib/string-format' );
var reduce = require( './main.js' );
// MAIN //
/**
* Return a function for performing a reduction over a list of specified dimensions in an input ndarray and assigning results to a provided output ndarray.
*
* @param {Function} fcn - reduction function
* @throws {TypeError} first argument must be a function
* @returns {Function} function for performing a reduction
*
* @example
* var Float64Array = require( '@stdlib/array-float64' );
* var filled = require( '@stdlib/array-base-filled' );
* var ndarray2array = require( '@stdlib/ndarray-base-to-array' );
* var base = require( '@stdlib/ndarray-base-every' );
*
* // Create data buffers:
* var xbuf = new Float64Array( [ 1.0, 2.0, 3.0, 4.0, 5.0, 0.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0 ] );
* var ybuf = filled( false, 3 );
*
* // Define the array shapes:
* var xsh = [ 1, 3, 2, 2 ];
* var ysh = [ 1, 3 ];
*
* // Define the array strides:
* var sx = [ 12, 4, 2, 1 ];
* var sy = [ 3, 1 ];
*
* // Define the index offsets:
* var ox = 0;
* var oy = 0;
*
* // Create an input ndarray-like object:
* var x = {
* 'dtype': 'float64',
* 'data': xbuf,
* 'shape': xsh,
* 'strides': sx,
* 'offset': ox,
* 'order': 'row-major'
* };
*
* // Create an output ndarray-like object:
* var y = {
* 'dtype': 'generic',
* 'data': ybuf,
* 'shape': ysh,
* 'strides': sy,
* 'offset': oy,
* 'order': 'row-major'
* };
*
* // Create a function for performing a reduction over subarrays:
* var every = factory( base );
* // returns <Function>
*
* // Perform a reduction:
* every( [ x, y ], [ 2, 3 ] );
*
* var arr = ndarray2array( y.data, y.shape, y.strides, y.offset, y.order );
* // returns [ [ true, false, true ] ]
*/
function factory( fcn ) {
if ( !isFunction( fcn ) ) {
throw new TypeError( format( 'invalid argument. First argument must be a function. Value: `%s`.', fcn ) );
}
return reducer;
/**
* Performs a reduction over a list of specified dimensions in an input ndarray and assigns results to a provided output ndarray.
*
* @private
* @param {ArrayLikeObject<Object>} arrays - array-like object containing ndarrays
* @param {IntegerArray} dims - list of dimensions over which to perform a reduction
* @param {Options} [options] - function options
* @returns {void}
*/
function reducer( arrays, dims, options ) {
var opts;
if ( arguments.length > 2 ) {
opts = options;
} else {
opts = {};
}
return reduce( fcn, arrays, dims, opts );
}
}
// EXPORTS //
module.exports = factory;