From 0f3b415b3177a6b73828c09090cfae3a9c6dd5c9 Mon Sep 17 00:00:00 2001 From: Peter Mikkelsen Date: Sat, 29 Jan 2022 00:09:53 +0000 Subject: =?UTF-8?q?Handle=20scalars=20in=20monadic=20=E2=89=A0=20and=20?= =?UTF-8?q?=E2=88=AA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- functions.c | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/functions.c b/functions.c index ed0fb36..2425798 100644 --- a/functions.c +++ b/functions.c @@ -369,6 +369,13 @@ fnSame(Array *right) Array * fnUniqueMask(Array *right) { + if(right->rank == 0){ + Array *res = allocarray(AtypeInt, 1, 1); + res->shape[0] = 1; + res->intdata[0] = 1; + return res; + } + return rundfn(L"(⍳≢⍵)≤⊃∘⍸¨↓≡⌾⍨⌷∘⍵¨⍳≢⍵", nil, nil, nil, right); } @@ -620,6 +627,9 @@ fnEnlist(Array *right) Array * fnUnique(Array *right) { + if(right->rank == 0) + return fnRavel(right); + return rundfn(L"(≠⍵)⌿⍵", nil, nil, nil, right); } -- cgit v1.2.3