wavenet.py 文件源码

def DilatedConvBlock(name, input_dim, output_dim, filter_size, inputs):
    result = inputs
    for i in xrange(DILATION_LEVEL):
        d = numpy.left_shift(2, i)
        result = DilatedConv1D(name+'Dilation'+str(d), DIM, DIM, 5, result, d, mask_type='b')
        result = relu(result)
    return result