keras-team / keras-core

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
Apache License 2.0
1.27k stars 117 forks source link

Split fix #789

Closed jackd closed 1 year ago

jackd commented 1 year ago

Split's compute_output_spec does not return the final element (though the underlying call does). This addresses this, plus allows passing tuples to keras.ops.split without error.