在损耗的计算中，我需要计算两个向量一维a和b的卷积。在numpy中，无需考虑批处理维度，可以通过np.convolve(a,b)轻松解决。但是，在张量流中，现在a和b都成为二维向量。例如，a是$ N x y $，b是$ N x y $，其中N是batch_size。
所以现在我想进行逐级卷积，可以实现以下操作：
for i in range(N):
z[i,:] = np.convolve(a[i.:], b[i,:])
有没有合适的API可以解决问题？我搜索了许多相关主题，但都与CNN网络有关。但是，我只想简单地计算与一维向量的分批卷积。
请帮助我，谢谢！