cubek-pool 0.3.0

CubeK: Pool
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

use thiserror::Error;

#[derive(Error, Debug, Clone)]
pub enum PoolError {
    #[error("Unsupported pooling mode: {mode}")]
    UnsupportedMode { mode: String },

    #[error("Invalid tensor rank: input {input} output {output}")]
    InvalidRank { input: usize, output: usize },

    #[error("Batch size mismatch: input has {input} but output has {output}")]
    BatchMismatch { input: usize, output: usize },

    #[error("Channel count mismatch: input has {input} but output has {output}")]
    ChannelMismatch { input: usize, output: usize },

    #[error("{tensor} spatial dimensions must be non-zero, got {actual:?}")]
    InvalidSpatialSize {
        tensor: &'static str,
        actual: Vec<usize>,
    },

    #[error("Output spatial shape mismatch: expected {expected:?} but got {actual:?}")]
    OutputSizeMismatch {
        expected: Vec<usize>,
        actual: Vec<usize>,
    },

    #[error("Input gradient shape mismatch: expected {expected:?} but got {actual:?}")]
    InputGradientShapeMismatch {
        expected: Vec<usize>,
        actual: Vec<usize>,
    },
}