Muhammad Tanvir

Results 2 issues of Muhammad Tanvir

This patch adds the required bit shift operation for `joint_matrix` which was accidentally skipped from #491

Only works for `trans_a = n` and `trans_b = n` case. Hasn't been tested for corner cases either.