Crate primitiv [−] [src]
Re-exports
pub use functions::node_funcs as node_functions; |
pub use functions::tensor_funcs as tensor_functions; |
Modules
| devices | |
| functions | |
| initializers | |
| optimizers |
Macros
| impl_model |
Structs
| Graph |
Computation graph. |
| Model |
Set of parameters and specific algorithms. |
| Node |
Pointer of a node in the computation graph. |
| Parameter |
Struct to manage a trainable tensor parameter. |
| Shape |
Data structure to represent the shape of the node. |
| Tensor |
Value with any dimensions. |
Traits
| Device |
|
| Initializer |
|
| ModelImpl | |
| Optimizer |
|
| Wrap |