search

jonasrauber / eagerpy

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

How to Transform a torch tensor to tensorflow tensor #18

Open kepengxu opened 3 years ago

jonasrauber commented 3 years ago

This is not currently the purpose of EagerPy. EagerPy, right now, gives you an API to write code that works the same way with PyTorch tensors and TensorFlow tensors, but not mixing them simultaneously. In principle, we might be able to add that if it is needed (keep in mind though that TensorFlow and PyTorch on the same GPU don't behave that well).