burn::tensor::quantization

Function params_shape

pub fn params_shape(data_shape: &Shape, level: QuantLevel) -> Shape
Expand description

Calculate the shape of the quantization parameters for a given tensor and level