Laurent Mazare 2a74019ec6 Vision dataset (#179) ...

* Add some readers for the mnist dataset.

* Import the cifar and mnist dataset.

2023-07-16 23:43:55 +01:00

..

vision

Vision dataset (#179)

2023-07-16 23:43:55 +01:00

activation.rs

[nn] Move the Embedding and Activation parts. (#116)

2023-07-10 10:24:52 +01:00

conv.rs

Add some documentation and test to the linear layer. (#151)

2023-07-12 20:24:23 +01:00

embedding.rs

Add some documentation and test to the linear layer. (#151)

2023-07-12 20:24:23 +01:00

init.rs

Sketch the tensor initialization module. (#134)

2023-07-11 07:41:46 +01:00

layer_norm.rs

Simplify the parameters used by sum and sum_keepdim. (#165)

2023-07-14 08:22:08 +01:00

lib.rs

Vision dataset (#179)

2023-07-16 23:43:55 +01:00

linear.rs

Add some documentation and test to the linear layer. (#151)

2023-07-12 20:24:23 +01:00

optim.rs

Add the gradient for reduce-sum. (#162)

2023-07-13 20:14:10 +01:00

var_builder.rs

Add backtrace information to errors where relevant. (#166)

2023-07-14 09:31:25 +01:00