def test(S, N, K):
    """ S: image size, N: num images, K: kernel size"""
a = np.random.randn(S, S, N)
b = np.random.randn(K, K)
valid = [slice(K // 2, -K // 2 + 1), slice(K // 2, -K // 2 + 1)]
свертка Python с другим измерением