Ejemplo n.º 1
0
 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
Ejemplo n.º 2
0
 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
Ejemplo n.º 3
0
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)