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

2023-07-14 08:22:08 +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

Add the SGD optimizer (#160)

2023-07-13 19:05:44 +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

Cleanup the main crate error and add a couple dedicated ones (#142)

2023-07-12 09:17:08 +01:00