emission.unsqueeze(2).expand(-1, -1, out_size, -1) + self.transition.unsqueeze(0)

[ttyuuuuuuuuuu]

数量少的时候相加正常得出结果，但是数量大的时候出现oom，这个时候batchsize已经调成2了，请问这个可以怎么解决？

数据集信息：

服务器信息（0卡）