diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index 4a3a2c6ab..d7bd9724e 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -34,10 +34,10 @@ class Tensor: no_grad: ClassVar[bool] = False default_type: ClassVar[DType] = dtypes.float32 - def __init__(self, data:Union[int, float, list, LazyBuffer, np.ndarray], device=Device.DEFAULT, dtype:Optional[DType]=None, requires_grad:Optional[bool]=None): + def __init__(self, data:Union[int, float, list, tuple, LazyBuffer, np.ndarray], device=Device.DEFAULT, dtype:Optional[DType]=None, requires_grad:Optional[bool]=None): assert dtype is None or isinstance(dtype, DType), f"invalid dtype {dtype}" device = Device.canonicalize(device) - if isinstance(data, list): + if isinstance(data, (list, tuple)): data = np.array(data, dtype=(dtype if dtype is not None else Tensor.default_type).np) if isinstance(data, LazyBuffer):