这里参考了大哥的测试代码(Linux服务器Mamba2安装及example运行问题-CSDN博客) (1)mamba测试 这个一般没啥问题 import torch from mamba_ssm import Mamba batch, length, dim = 2, 64, 16 x = torch.randn(batch, length, dim).to("cuda") model = Mamba( # This module uses roughly 3 * expand * ...