OnesLike

Returns a tensor filled with the scalar value 1, with the same size as input.

Abstract Signature:

OnesLike(input: Tensor, dtype: DType = None)

PyTorch

API: torch.ones_like
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.ones_like
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.ones_like
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.ones_like
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.ones_like
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.ones_like
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.ones_like
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.numpy.ones_like
Strategy: Direct Mapping
Official Docs ↗