From be9d624b581d86ae9261c8715419e2c358ef1712 Mon Sep 17 00:00:00 2001 From: Athan Reines Date: Thu, 23 Nov 2023 01:25:54 -0800 Subject: [PATCH] feat: add support for accessor strided arrays --- .../@stdlib/array/base/strided2array4d/lib/main.js | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/lib/node_modules/@stdlib/array/base/strided2array4d/lib/main.js b/lib/node_modules/@stdlib/array/base/strided2array4d/lib/main.js index b2892518f65..a88e9dc2713 100644 --- a/lib/node_modules/@stdlib/array/base/strided2array4d/lib/main.js +++ b/lib/node_modules/@stdlib/array/base/strided2array4d/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 strided2array4d( x, shape, strides, offset ) { + var get; var out; var dx0; var dx1; @@ -66,6 +72,8 @@ function strided2array4d( x, shape, strides, offset ) { var t2; var t1; + get = resolveGetter( x ); + S3 = shape[ 0 ]; S2 = shape[ 1 ]; S1 = shape[ 2 ]; @@ -87,7 +95,7 @@ function strided2array4d( 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 );