Getting the locations of the blocks returned by fold

1

Hi there, my question is the following:

let’s assume I call nn.functional.fold on a tensor v of shape (b,c,w,h). It returns me a rank-3 tensor of shape (b, blocksize, blockcount). I’m interested in knowing the actual position for each block within the input v. Is there an easy way ?

thanks