Auto-generated commit

stdlib-js · Oct 7, 2023 · 638f254 · 638f254
1 parent 9d541c4
commit 638f254
Show file tree

Hide file tree

Showing 13 changed files with 759 additions and 5 deletions.
diff --git a/base/lib/index.js b/base/lib/index.js
@@ -499,6 +499,15 @@ setReadOnly( ns, 'removeSingletonDimensions', require( './../../base/remove-sing
 */
 setReadOnly( ns, 'serializeMetaData', require( './../../base/serialize-meta-data' ) );
 
+/**
+* @name shape
+* @memberof ns
+* @readonly
+* @type {Function}
+* @see {@link module:@stdlib/ndarray/base/shape}
+*/
+setReadOnly( ns, 'shape', require( './../../base/shape' ) );
+
 /**
 * @name shape2strides
 * @memberof ns

diff --git a/base/shape/README.md b/base/shape/README.md
@@ -0,0 +1,160 @@
+<!--
+
+@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.
+
+-->
+
+# shape
+
+> Return the shape of a provided [ndarray][@stdlib/ndarray/base/ctor].
+
+<!-- Section to include introductory text. Make sure to keep an empty line after the intro `section` element and another before the `/section` close. -->
+
+<section class="intro">
+
+</section>
+
+<!-- /.intro -->
+
+<!-- Package usage documentation. -->
+
+<section class="usage">
+
+## Usage
+
+```javascript
+var shape = require( '@stdlib/ndarray/base/shape' );
+```
+
+#### shape( x, copy )
+
+Returns the shape of a provided [ndarray][@stdlib/ndarray/base/ctor].
+
+```javascript
+var zeros = require( '@stdlib/ndarray/zeros' );
+
+var x = zeros( [ 3, 2, 3 ] );
+// returns <ndarray>
+
+var sh = shape( x, false );
+// returns [ 3, 2, 3 ]
+```
+
+When `copy` is `false`, changes to the returned shape array may mutate the input [ndarray][@stdlib/ndarray/base/ctor] shape. If there is a chance that the returned shape will be mutated (either directly or by downstream consumers), set `copy` to `true` to prevent unintended side effects.
+
+```javascript
+var zeros = require( '@stdlib/ndarray/zeros' );
+
+var x = zeros( [ 3, 2, 3 ] );
+// returns <ndarray>
+
+var sh = shape( x, true );
+// returns [ 3, 2, 3 ]
+
+var bool = ( x.shape === sh );
+// returns false
+```
+
+</section>
+
+<!-- /.usage -->
+
+<!-- Package usage notes. Make sure to keep an empty line after the `section` element and another before the `/section` close. -->
+
+<section class="notes">
+
+</section>
+
+<!-- /.notes -->
+
+<!-- Package usage examples. -->
+
+<section class="examples">
+
+## Examples
+
+<!-- eslint no-undef: "error" -->
+
+<!-- eslint-disable new-cap -->
+
+```javascript
+var zeros = require( '@stdlib/ndarray/zeros' );
+var slice = require( '@stdlib/ndarray/slice' );
+var E = require( '@stdlib/slice/multi' );
+var S = require( '@stdlib/slice/ctor' );
+var shape = require( '@stdlib/ndarray/base/shape' );
+
+// Create an array:
+var x = zeros( [ 10, 10, 10, 10 ] );
+// returns <ndarray>
+
+// Define some slices:
+var slices = [
+    // :,:,:,:
+    E( null, null, null, null ),
+
+    // 5:10,4,2:4,::-1
+    E( S( 5, 10 ), 4, S( 2, 4 ), S( null, null, -1 ) ),
+
+    // :,:,2,:
+    E( null, null, 2, null ),
+
+    // 1,2,3,:
+    E( 1, 2, 3, null ),
+
+    // 1,3,::2,4::2
+    E( 1, 3, S( null, null, 2 ), S( 4, null, 2 ) )
+];
+
+// Determine the shape of each slice...
+var s;
+var i;
+for ( i = 0; i < slices.length; i++ ) {
+    s = slice( x, slices[ i ] );
+    console.log( 'slice(%s) => %s', x.shape.join( 'x' ), shape( s, false ).join( 'x' ) );
+}
+```
+
+</section>
+
+<!-- /.examples -->
+
+<!-- Section to include cited references. If references are included, add a horizontal rule *before* the section. Make sure to keep an empty line after the `section` element and another before the `/section` close. -->
+
+<section class="references">
+
+</section>
+
+<!-- /.references -->
+
+<!-- Section for related `stdlib` packages. Do not manually edit this section, as it is automatically populated. -->
+
+<section class="related">
+
+</section>
+
+<!-- /.related -->
+
+<!-- Section for all links. Make sure to keep an empty line after the `section` element and another before the `/section` close. -->
+
+<section class="links">
+
+[@stdlib/ndarray/base/ctor]: https://github.com/stdlib-js/ndarray/tree/main/base/ctor
+
+</section>
+
+<!-- /.links -->
diff --git a/base/shape/benchmark/benchmark.js b/base/shape/benchmark/benchmark.js
@@ -0,0 +1,82 @@
+/**
+* @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.
+*/
+
+'use strict';
+
+// MODULES //
+
+var bench = require( '@stdlib/bench' );
+var zeros = require( './../../../zeros' );
+var isCollection = require( '@stdlib/assert/is-collection' );
+var pkg = require( './../package.json' ).name;
+var shape = require( './../lib' );
+
+
+// MAIN //
+
+bench( pkg+':copy=false', function benchmark( b ) {
+	var values;
+	var out;
+	var i;
+
+	values = [
+		zeros( [ 10, 10, 10, 1 ] ),
+		zeros( [ 5, 5, 5, 1, 1 ] ),
+		zeros( [ 3, 4, 5 ] )
+	];
+
+	b.tic();
+	for ( i = 0; i < b.iterations; i++ ) {
+		out = shape( values[ i%values.length ], false );
+		if ( typeof out !== 'object' ) {
+			b.fail( 'should return an array' );
+		}
+	}
+	b.toc();
+	if ( !isCollection( out ) ) {
+		b.fail( 'should return an array' );
+	}
+	b.pass( 'benchmark finished' );
+	b.end();
+});
+
+bench( pkg+':copy=true', function benchmark( b ) {
+	var values;
+	var out;
+	var i;
+
+	values = [
+		zeros( [ 10, 10, 10, 1 ] ),
+		zeros( [ 5, 5, 5, 1, 1 ] ),
+		zeros( [ 3, 4, 5 ] )
+	];
+
+	b.tic();
+	for ( i = 0; i < b.iterations; i++ ) {
+		out = shape( values[ i%values.length ], true );
+		if ( typeof out !== 'object' ) {
+			b.fail( 'should return an array' );
+		}
+	}
+	b.toc();
+	if ( !isCollection( out ) ) {
+		b.fail( 'should return an array' );
+	}
+	b.pass( 'benchmark finished' );
+	b.end();
+});
diff --git a/base/shape/docs/repl.txt b/base/shape/docs/repl.txt
@@ -0,0 +1,30 @@
+
+{{alias}}( x, copy )
+    Returns the shape of a provided ndarray.
+
+    Parameters
+    ----------
+    x: ndarray
+        Input ndarray.
+
+    copy: boolean
+        Boolean indicating whether to explicitly copy the value assigned to the
+        input ndarray's `shape` property. When `copy` is `false`, changes to the
+        returned shape array may mutate the input ndarray shape. If there is a
+        chance that the returned shape will be mutated (either directly or by
+        downstream consumers), set `copy` to `true` to prevent unintended side
+        effects.
+
+    Returns
+    -------
+    out: Array<integer>
+        Shape.
+
+    Examples
+    --------
+    > var out = {{alias}}( {{alias:@stdlib/ndarray/zeros}}( [ 3, 3, 3 ] ), false )
+    [ 3, 3, 3 ]
+
+    See Also
+    --------
+
diff --git a/base/shape/docs/types/index.d.ts b/base/shape/docs/types/index.d.ts
@@ -0,0 +1,47 @@
+/*
+* @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 { ndarray, Shape } from '@stdlib/types/ndarray';
+
+/**
+* Returns the shape of a provided ndarray.
+*
+* ## Notes
+*
+* -   When `copy` is `false`, changes to the returned shape array may mutate the input ndarray shape. If there is a chance that the returned shape will be mutated (either directly or by downstream consumers), set `copy` to `true` to prevent unintended side effects.
+*
+* @param x - input ndarray
+* @param copy - boolean indicating whether to explicitly copy the value assigned to the input ndarray's `shape` property
+* @returns shape
+*
+* @example
+* var zeros = require( `@stdlib/ndarray/zeros` );
+*
+* var sh = shape( zeros( [ 3, 3, 3 ] ), false );
+* // returns [ 3, 3, 3 ]
+*/
+declare function shape( x: ndarray, copy: boolean ): Shape;
+
+
+// EXPORTS //
+
+export = shape;
diff --git a/base/shape/docs/types/test.ts b/base/shape/docs/types/test.ts
@@ -0,0 +1,60 @@
+/*
+* @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.
+*/
+
+import zeros = require( './../../../../zeros' );
+import shape = require( './index' );
+
+
+// TESTS //
+
+// The function returns an ndarray shape...
+{
+	shape( zeros( [ 3, 2, 1 ] ), false ); // $ExpectType Shape
+	shape( zeros( [ 3, 2, 1 ] ), true ); // $ExpectType Shape
+}
+
+// The compiler throws an error if the function is provided a first argument which is not an ndarray...
+{
+	shape( '5', false ); // $ExpectError
+	shape( 5, false ); // $ExpectError
+	shape( true, false ); // $ExpectError
+	shape( false, false ); // $ExpectError
+	shape( null, false ); // $ExpectError
+	shape( undefined, false ); // $ExpectError
+	shape( [ '1', '2' ], false ); // $ExpectError
+	shape( {}, false ); // $ExpectError
+	shape( ( x: number ): number => x, false ); // $ExpectError
+}
+
+// The compiler throws an error if the function is provided a second argument which is not a boolean...
+{
+	shape( zeros( [ 3, 2, 1 ] ), '5' ); // $ExpectError
+	shape( zeros( [ 3, 2, 1 ] ), 5 ); // $ExpectError
+	shape( zeros( [ 3, 2, 1 ] ), null ); // $ExpectError
+	shape( zeros( [ 3, 2, 1 ] ), undefined ); // $ExpectError
+	shape( zeros( [ 3, 2, 1 ] ), [ '1', '2' ] ); // $ExpectError
+	shape( zeros( [ 3, 2, 1 ] ), {} ); // $ExpectError
+	shape( zeros( [ 3, 2, 1 ] ), ( x: number ): number => x ); // $ExpectError
+}
+
+// The compiler throws an error if the function is provided an unsupported number of arguments...
+{
+	shape(); // $ExpectError
+	shape( zeros( [ 2, 2 ] ) ); // $ExpectError
+	shape( zeros( [ 2, 2 ] ), false, {} ); // $ExpectError
+}