set_parallel_threshold

set_parallel_threshold(threshold)

Set the output-element threshold where kernels switch to Rayon.

Parameters

threshold : int

Positive output-element count. For a stack shaped (N, H, W), this is compared with H * W. For arbitrary N-D stacks, this is compared with np.prod(arr.shape[1:]).