From 6ece4597dcd71ffd2176bfbd7e39e22a6aa40b9e Mon Sep 17 00:00:00 2001 From: Athan Reines Date: Thu, 23 Nov 2023 01:26:57 -0800 Subject: [PATCH] feat: add support for accessor strided arrays --- .../@stdlib/array/base/strided2array5d/lib/main.js | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/lib/node_modules/@stdlib/array/base/strided2array5d/lib/main.js b/lib/node_modules/@stdlib/array/base/strided2array5d/lib/main.js index b47d086fb06..b58b054181f 100644 --- a/lib/node_modules/@stdlib/array/base/strided2array5d/lib/main.js +++ b/lib/node_modules/@stdlib/array/base/strided2array5d/lib/main.js @@ -18,6 +18,11 @@ 'use strict'; +// MODULES // + +var resolveGetter = require( '@stdlib/array/base/resolve-getter' ); + + // MAIN // /** @@ -46,6 +51,7 @@ * // returns [ [ [ [ [ 1, 4 ], [ 2, 5 ], [ 3, 6 ] ] ] ] ] */ function strided2array5d( x, shape, strides, offset ) { + var get; var out; var dx0; var dx1; @@ -71,6 +77,8 @@ function strided2array5d( x, shape, strides, offset ) { var t2; var t1; + get = resolveGetter( x ); + S4 = shape[ 0 ]; S3 = shape[ 1 ]; S2 = shape[ 2 ]; @@ -97,7 +105,7 @@ function strided2array5d( x, shape, strides, offset ) { t1 = []; ix0 = ix1 + ( dx1*i1 ); for ( i0 = 0; i0 < S0; i0++ ) { - t1.push( x[ ix0 ] ); + t1.push( get( x, ix0 ) ); ix0 += dx0; } t2.push( t1 );