Does torch.matmul have arguments similar to LDA (leading dimension A) and LDB (leading dimension B)?
I’ve been working out some slicing solutions to get the same result, but this old code is poorly documented and I’d like to stay as close as possible to the original to avoid introducing bugs where I don’t understand the original intent.