-
-
Notifications
You must be signed in to change notification settings - Fork 804
/
Copy pathindex.js
62 lines (51 loc) · 1.78 KB
/
index.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
/**
* @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';
var base = require( '@stdlib/stats/base/ndarray/max' );
var uniform = require( '@stdlib/random/array/uniform' );
var dtypes = require( '@stdlib/ndarray/dtypes' );
var dtype = require( '@stdlib/ndarray/dtype' );
var ndarray2array = require( '@stdlib/ndarray/to-array' );
var ndarray = require( '@stdlib/ndarray/ctor' );
var UnaryStridedDispatch = require( './../lib' );
// Define the supported input and output data types:
var idt = dtypes( 'real_and_generic' );
var odt = dtypes( 'real_and_generic' );
// Define the policy mapping an input data type to an output data type:
var policy = 'same';
// Define a dispatch table:
var table = {
'default': base
};
// Create an interface for performing a reduction:
var max = new UnaryStridedDispatch( table, [ idt ], odt, policy );
// Generate an array of random numbers:
var xbuf = uniform( 100, -1.0, 1.0, {
'dtype': 'generic'
});
// Wrap in an ndarray:
var x = new ndarray( 'generic', xbuf, [ 10, 10 ], [ 10, 1 ], 0, 'row-major' );
// Perform a reduction:
var y = max.apply( x, {
'dims': [ 0 ]
});
// Resolve the output array data type:
var dt = dtype( y );
console.log( dt );
// Print the results:
console.log( ndarray2array( y ) );