search

jonasrauber / eagerpy

PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
https://eagerpy.jonasrauber.de
MIT License
694 stars 39 forks source link

rotation pytorch #4

Closed lukas-schott closed 4 years ago

lukas-schott commented 4 years ago

added rotation for pytorch with same results and similar api as tensorflow_addons.image.rotate(images, angles, interpolation='BILINEAR')

jonasrauber commented 4 years ago

I don't think we'll add this to the Tensor class, too much code to maintain in EagerPy and too specialized. Let's make it part of the attack for now, until we need it in more than one place.

jonasrauber commented 4 years ago

e.g. a rotate.py module with a rotate function that takes an EagerPy tensor and (depending on the subclass) calls a framework-specific rotate-implementation