params_shape

burn_tensor::quantization

Function params_shape 

Source 
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