I am not sure if it is possible yet. As a naive solution (assuming that computation is not a bottleneck), why can’t you multiply the conv output with a mask that specifies 1 in the locations selected by user or other algorithms?