(f)
| 28 | |
| 29 | |
| 30 | def _wrap_masked(f): |
| 31 | @wraps(f) |
| 32 | def _(a, value): |
| 33 | a = asanyarray(a) |
| 34 | value = asanyarray(value) |
| 35 | ainds = tuple(range(a.ndim))[::-1] |
| 36 | vinds = tuple(range(value.ndim))[::-1] |
| 37 | oinds = max(ainds, vinds, key=len) |
| 38 | return blockwise(f, oinds, a, ainds, value, vinds, dtype=a.dtype) |
| 39 | |
| 40 | return _ |
| 41 | |
| 42 | |
| 43 | masked_greater = _wrap_masked(np.ma.masked_greater) |