diff --git a/pygrad/tensor.py b/pygrad/tensor.py index 0f0cc8b..400a2a8 100644 --- a/pygrad/tensor.py +++ b/pygrad/tensor.py @@ -6,8 +6,7 @@ def tensor(*args, **kwargs): class Tensor: # TODO Implement 'requires_grad' functionality. def __init__(self, value): - # NOTE We technically could support both numpy arrays and scalar values, - # but it is too much work. + # TODO Add support for scalar values. if not isinstance(value, np.ndarray): print(f"{type(value)} is not compatible with {np.ndarray}") exit(-1)