Array API compatible wrapper for :py:func:`np.ones_like `. See its docstring for more information.
(
x: Array, /, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None
)
| 273 | |
| 274 | |
| 275 | def ones_like( |
| 276 | x: Array, /, *, dtype: Optional[Dtype] = None, device: Optional[Device] = None |
| 277 | ) -> Array: |
| 278 | """ |
| 279 | Array API compatible wrapper for :py:func:`np.ones_like <numpy.ones_like>`. |
| 280 | |
| 281 | See its docstring for more information. |
| 282 | """ |
| 283 | from ._array_object import Array |
| 284 | |
| 285 | _check_valid_dtype(dtype) |
| 286 | if device not in ["cpu", None]: |
| 287 | raise ValueError(f"Unsupported device {device!r}") |
| 288 | return Array._new(np.ones_like(x._array, dtype=dtype)) |
| 289 | |
| 290 | |
| 291 | def tril(x: Array, /, *, k: int = 0) -> Array: |