Function convolve

tensorflow_agent/layers.py:13–14

(i, k)

    assert c_o % group == 0

    def convolve(i, k):
        return tf.nn.conv2d(i, k, [1, s_h, s_w, 1], padding=padding)

    if group == 1:
        conv = convolve(input, kernel)