Extract elements from matrix given indices from max_pool2d

[kshitij]

Probably this might help