Malloc for kernel output failed, Memory isnt enough

报错信息：

报错代码：

具体描述：

      在标红那行代码的前两行代码都执行过报错处的代码，但是均顺利执行，但运行到标红处的时候，调用 get_time_embdding_A函数执行到报错代码处，就出现了报错信息中的错误，这种情况是怎么回事呢？又该如何解决呢？

问题在将batch size设得更小以后解决了