MLXBackend.where#

MLXBackend.where(*args, **kwargs)#

Return elements from input depending on condition.

Parameters:
conditionBackendArray

Binary condition array.

*argsBackendArray

Values to choose from.

Returns:
BackendArray

Array of elements according to condition.