jax.numpy.ones#
- jax.numpy.ones(shape, dtype=None, *, device=None)[源代码]#
创建一个充满 1 的数组。
JAX 对
numpy.ones()
的实现。- 参数:
- 返回值:
具有指定形状和数据类型的数组,如果指定,则位于指定的设备上。
- 返回类型:
示例
>>> jnp.ones(4) Array([1., 1., 1., 1.], dtype=float32) >>> jnp.ones((2, 3), dtype=bool) Array([[ True, True, True], [ True, True, True]], dtype=bool)