Hmm I’m not sure if I can remove the for loop though, as I’m trying to retrieve remove a certain row in the m_alpha_beta and m_beta_alpha matrix. Stacking sounds like it’s worth a try though, I probably could just create a tensor and keep stacking them?