def array2string(a, max_line_width=None, precision=None, suppress_small=None, separator=' ', array_output=0): if array_output: prefix="array(" style=repr else: prefix = "" style=str return _array2string(a, max_line_width, precision, suppress_small, separator, prefix, style)
def array2string(array, **params): """ makes sure we get the same doctest with numpy and jax.numpy """ return _array2string(array, **dict(params, separator=', '))\ .replace('[ ', '[').replace(' ', ' ')