def sym_CorrMM(input, filters): # define theano graph and function input.name = 'input' filters.name = 'filters' rval = corr.CorrMM(border_mode, subsample)(input, filters) rval.name = 'corr_output' return rval
def sym_CorrMM(input, filters): # define theano graph and function input.name = "input" filters.name = "filters" rval = corr.CorrMM(border_mode, subsample, filter_dilation)(input, filters) rval.name = "corr_output" return rval
def conv_corr(inputs, filters, border_mode="valid", subsample=(1, 1), conv_mode='conv'): if conv_mode == 'conv': filters = filters[:, :, ::-1, ::-1] return corr.CorrMM(border_mode, subsample)(inputs, filters)