burn-rmexp-dyntensor 0.0.5

burn remote experiment dyntensor
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

//! # Common Error Types
use burn::prelude::Shape;
use burn::tensor::Slice;

/// Errors that can occur when checking tensor slices.
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum SlicingError {
    OutOfBounds {
        msg: String,
        shape: Shape,
        slices: Vec<Slice>,
    },
    InvalidRank {
        msg: String,
        shape: Shape,
        slices: Vec<Slice>,
    },
}

/// Common Errors.
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum DynTensorError {
    /// An error occurred while slicing.
    SliceError(SlicingError),

    /// Invalid Arguments.
    InvalidArgument { msg: String },

    /// The tensor rank is not supported for the requested operation.
    UnsupportedRank { msg: String, rank: usize },
}