Function fn

backends/mlx/test/test_ops.py:4105–4113

(shape, dtype)

    """

    def fn(shape, dtype):
        base = (
            torch.rand(shape, dtype=dtype)
            if uniform
            else torch.randn(shape, dtype=dtype)
        )
        if abs:
            base = base.abs()
        return (base * scale + offset,)

    return fn