renamed def to get_matrix_along_axis. Will now return a t matrix