Function affine 

fn affine(
    source: &Tensor,
    codes: StoredTensor,
    scales: StoredTensor,
    biases: StoredTensor,
    bits: u8,
) -> Result<Tensor>