I\'m trying to implement a little differently this guide here: https://nextjournal.com/gkoehler/pytorch-neural-style-transfer
but when I try to implement the get_feat