| # Licensed under the Apache License v2.0 with LLVM Exceptions. | |
| # See https://llvm.org/LICENSE.txt for license information. | |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | |
| import jax.numpy as jnp | |
| import numpy as np | |
| a = np.arange(15, dtype=np.float32).reshape(3, 5) | |
| a = a.T | |
| transpose = jnp.array(a) | |
| transpose_copy = jnp.array(a.copy()) | |
| assert (transpose == transpose_copy).all() |