2 回答

Using torch.matmul in pytorch prompts a mismatch of multiplication inputs
AI人工智能
yijun.deng yijun.deng 回答时间： 2025-3-5 16:45

Hello, The tool has treated the dimensions to be H, W, C, but actually your first dimension should be batch, I suggest you to use 2D Matmul. Best Regards, Yijun

赞0