MxNet, check if all elements of array have same value

Gluon
1

I need to check if two MxNet arrays are the same, so I’m using:

def check_equity(x1, x2):
    comp = mx.nd.equal(x1, x2)

which produces an array of 0/1 based on whether the corresponding values of x1 and x2 are equal. But how can I check if all the values of the produced comp array are 1s?

The lame way is to just iterate through them (they are images, so (3x224x224 arrays):

check = True
for ch in comp:
    for row in ch:
        for col in row:
            if col == 0:
                check = False
return check

I imagine there has to be some sort of .all() method, just can’t find any documentation about it.