From 17e97bbed7abc59fb754cf463e825e890b16815d Mon Sep 17 00:00:00 2001 From: Peter Mikkelsen Date: Wed, 26 Jan 2022 15:09:20 +0000 Subject: =?UTF-8?q?Implement=20replicate=20first=20and=20last=20=E2=8C=BF?= =?UTF-8?q?=20/?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- array.c | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'array.c') diff --git a/array.c b/array.c index e333870..3ec8639 100644 --- a/array.c +++ b/array.c @@ -100,17 +100,18 @@ scalarextend(Array *a, Array *b, Array **aa, Array **bb) aa and bb are unchanged. */ - if(a->size == 1 && b->size != 1){ + if(a->rank == 0 && b->rank != 0){ *aa = extend(a, b); *bb = fnSame(b); - }else if(b->size == 1 && a->size != 1){ + }else if(a->rank != 0 && b->rank == 0){ *aa = fnSame(a); *bb = extend(b, a); }else if(a->size == b->size && a->rank == b->rank){ /* Check that each dimension matches */ - for(int i = 0; i < a->rank; i++) + for(int i = 0; i < a->rank; i++){ if(a->shape[i] != b->shape[i]) return 0; + } *aa = fnSame(a); *bb = fnSame(b); }else -- cgit v1.2.3