Split CNN Feature Map

vision
2

You could use F.fold/.unfold as seen e.g. here.

1 Like