github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-18 18:35:12 -05:00
Code Issues Packages Projects Releases Wiki Activity

refactor: no len if it is not needed (#1598)

Browse Source 
Co-authored-by: Roelof van Dijk <roelof.van.dijk@vitestro.com>
This commit is contained in:
Roelof van Dijk
2023-08-21 23:06:32 +02:00
committed by GitHub
parent 2c8f8ac611
commit 109100656f
11 changed files with 19 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/tensor.py
View File

@@ -282,7 +282,7 @@ class Tensor:
raise IndexError(f"too many indices for tensor of dimension {len(self.shape)}")
ellipses_found = [i for i, v in enumerate(orig_slices) if v is Ellipsis]
if len(ellipses_found) > 1: raise IndexError("an index can only have a single ellipsis ('...')")
ellipsis_idx = len(orig_slices) if len(ellipses_found) == 0 else ellipses_found[0]
ellipsis_idx = ellipses_found[0] if ellipses_found else len(orig_slices)
orig_slices[ellipsis_idx:ellipsis_idx+1] = [slice(None)] * (len(self.shape) - num_slices)
tensor_found = [(i,v) for i, v in enumerate(orig_slices) if isinstance(v, Tensor)]