Shards the tensor across the given devices in place.
(self, devices:tuple[str, ...], axis:int|None=None)
| 397 | return Tensor(uop).is_param_(self.is_param) |
| 398 | |
| 399 | def shard_(self, devices:tuple[str, ...], axis:int|None=None) -> Tensor: |
| 400 | """ |
| 401 | Shards the tensor across the given devices in place. |
| 402 | """ |
| 403 | return self.replace(self.shard(devices, axis)) |
| 404 | |
| 405 | def shard_like(self, y:Tensor) -> Tensor: |
| 406 | """ |