ruda-tensor 0.21.49

Ruda tensor operations, backend contracts, and framework-facing primitives.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

/// Module with convolution operations.
pub mod conv;

/// Module with linear operations.
pub mod linear;

pub mod embedding;

/// Module with attention operations.
pub mod attention;

/// Module with CTC loss operations.
pub mod ctc;

/// Module with unfold operations.
pub mod unfold;

/// Module with pooling operations.
pub mod pool;
/// Native-dimensional interpolation using the backend's original spatial filters.
pub mod interpolation;

/// Saved-statistics normalization and complete first-order gradients.
pub mod normalization;
/// Actual channel-group normalization and selected first-order derivatives.
pub mod group_normalization;

/// Saved-working-output softmax and log-softmax training on the current backend.
pub mod softmax;

/// Working-storage activation forward and first-order training operations.
pub mod activation_training;
/// Shared/channel-wise PReLU forward and independently selected training derivatives.
pub mod prelu_training;

/// Module for grid_sample operations
pub mod grid_sample;

mod base;

pub use base::*;