(self, stop, start=0, step=1, type_as=None)
| 1569 | return self._change_device(jnp.ones(shape, dtype=type_as.dtype), type_as) |
| 1570 | |
| 1571 | def arange(self, stop, start=0, step=1, type_as=None): |
| 1572 | return jnp.arange(start, stop, step) |
| 1573 | |
| 1574 | def full(self, shape, fill_value, type_as=None): |
| 1575 | if type_as is None: |