From 22f6c7e5b3fd45480f2ef7827474fdb9d6c6f82c Mon Sep 17 00:00:00 2001 From: Raghuram Subramani Date: Wed, 29 May 2024 18:11:05 +0530 Subject: Value -> Scalar --- src/scalar.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 src/scalar.py (limited to 'src/scalar.py') diff --git a/src/scalar.py b/src/scalar.py new file mode 100644 index 0000000..a67b7ae --- /dev/null +++ b/src/scalar.py @@ -0,0 +1,20 @@ +class Scalar: + def __init__(self, data, _children=(), _op='', label='') -> None: + self.label = label + + self.data = float(data) + self.grad = 0.0 + + self._prev = set(_children) + self._op = _op + + def __repr__(self) -> str: + return f'Scalar({self.data})' + + def __add__(self, y): + result = self.data + y.data + return Scalar(result, (self, y), _op='+') + + def __mul__(self, y): + result = self.data * y.data + return Scalar(result, (self, y), _op='*') -- cgit v1.2.3