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