albanD / subclass_zoo

126 stars 23 forks source link

In complex numbers example, mm sample impl seems just regular complex number multiplication impl and not mm #53

Open vadimkantorov opened 2 weeks ago

vadimkantorov commented 2 weeks ago

https://github.com/albanD/subclass_zoo/blob/main/complex_tensor.py#L27-L31

right?

albanD commented 2 weeks ago

Ho yes! Can you send a PR adding a bigger shape to the Tensors and fixing the mm?