Max-pooling with complex masks in PyTorch

Question

Suppose I have a matrix src with shape (5, 3) and a boolean matrix adj with shape (5, 5) as follow,

src = ten