# 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

ones_splat =  jax.numpy.ones((3, 4))
print(ones_splat)

twos = ones_splat + ones_splat
print(twos)

ones_degenerate = jax.numpy.ones((4, 0))
twos_degenerate = ones_degenerate + ones_degenerate
print(twos_degenerate)