Commit graph

5 commits

Author SHA1 Message Date
lucasdelimanogueira
418bb20b29 fix add broadcasted backward grad wrong values 2024-05-18 15:08:31 -03:00
lucasdelimanogueira
5dd3f3d202 implementing keepdim sum 2024-05-17 19:08:09 -03:00
lucasdelimanogueira
de6f2af637 fix upgrade grads reference to parameter 2024-05-06 20:40:06 -03:00
lucasdelimanogueira
e5819a4121 nn module v1 2024-05-06 19:47:27 -03:00
lucasdelimanogueira
1e00930c4e Fix nn.Module linear 2024-05-06 13:23:22 -03:00