Laurent Mazare 1f26042693 Move some shared functions to the nn module. (#221)

2023-07-22 13:25:11 +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

Rename the .r functions to .dims so as to be a bit more explicit. (#220)

2023-07-22 10:39:27 +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

Rename the .r functions to .dims so as to be a bit more explicit. (#220)

2023-07-22 10:39:27 +01:00

lib.rs

Move some shared functions to the nn module. (#221)

2023-07-22 13:25:11 +01:00

linear.rs

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

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

loss.rs

Move some shared functions to the nn module. (#221)

2023-07-22 13:25:11 +01:00

ops.rs

Move some shared functions to the nn module. (#221)

2023-07-22 13:25:11 +01:00

optim.rs

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

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

var_builder.rs

[Proposal] Remove SafeTensor wrapper (allows finer control for users).

2023-07-19 16:25:44 +02:00