Skip to main content

transpose_linear_weights

Function transpose_linear_weights 

Source
pub fn transpose_linear_weights(data: &TensorData) -> TensorData
Expand description

Transpose weights if needed for format conversion.

PyTorch Linear: [out_features, in_features] Some frameworks: [in_features, out_features]