Hi there!
I am sort of a newbie with tensor manipulation, I have looked around for this, but I can’t seem to get anywhere, and I am looking for something using torch tensors that would allow me to do the following:
results = []
for xs in x:
    r = []
    for xss in xs:
        r.append(sift(xss))
    results.append(r)

Where x is a tensor, I am just going through its batch_size and feature_channels dimensions in these loops and applying a SIFT layer for each image patch group.
I would also like to be able to run this as fast as possible
Thanks in advance!