如何在cntk中使用卷积函数？

Question

我试图使用cntk.convolution函数对两个常量数组进行卷积。

这是我的代码：

import cntk as C

w = C.constant(1, (2, 2))
a = C.constant(2, (2, 2))

c = C.convolution(w, a)

print(c.eval())

但它会导致以下错误

  

RuntimeError：卷积当前要求主操作数具有动态轴

我应该如何使用convolution功能？
感谢。

Answer 1

我在卷积方法的docstring中找到了这个例子。

img = np.reshape(np.arange(25.0, dtype=np.float32), (1, 5, 5))
x = C.input_variable(img.shape)
filter = np.reshape(np.array([2, -1, -1, 2], dtype=np.float32), (1, 2, 2))
kernel = C.constant(value=filter)
out = np.round(C.convolution(kernel, x, auto_padding=[False]).eval({x: [img]}), 5)
print(out)