EinopsError: Error while processing rearrange-reduction pattern "(b f) d c -> (b d) f c". Input tensor shape: torch.Size([2, 4096, 320]). Additional info: {'f': 16}. Shape mismatch, can't divide axis of length 2 in chunks of 16
用时:10.6 sec.
A: 2.52 GB, R: 2.62 GB, Sys: 3.7/6 GB (61.0%)