How to split tensor to small chunk for matrix multiplication to solve memory issue

Question

I have two matrix: torch.Size([1, 8, 50176, 64]) torch.Size([1, 8, 64, 50176]) I need to perform matrix mutiplicatio to get torch size [1, 8, 50176, 50176], but I got cuda o