(self, shape, fill_value, type_as=None)
| 2525 | return cp.arange(start, stop, step) |
| 2526 | |
| 2527 | def full(self, shape, fill_value, type_as=None): |
| 2528 | if isinstance(shape, (list, tuple)): |
| 2529 | shape = tuple(int(i) for i in shape) |
| 2530 | if type_as is None: |
| 2531 | return cp.full(shape, fill_value) |
| 2532 | else: |
| 2533 | with cp.cuda.Device(type_as.device): |
| 2534 | return cp.full(shape, fill_value, dtype=type_as.dtype) |
| 2535 | |
| 2536 | def eye(self, N, M=None, type_as=None): |
| 2537 | if type_as is None: |