При вычислении потерь мне нужно вычислить свертку двух векторов одномерных a и b.В целом, без учета размера партии, она может быть легко решена с помощью np.convolve(a,b).Однако в тензорном потоке теперь a и b становятся двумерным вектором.Например, a равно $ N \ times x $, b равно $ N \ times y $, где N - размер пакета.
Итак, теперь я хочу выполнить пакетную свертку, которая может выполнить следующую операцию:
for i in range(N):
z[i,:] = np.convolve(a[i.:], b[i,:])
Есть ли подходящий API, способный решить проблему?Я искал много связанных тем, но все о сетях CNN.Тем не менее, я хочу просто вычислить пакетное свертывание с одномерными векторами.
Пожалуйста, помогите мне и заранее спасибо