What's the best way to implement Targeted Fast Gradient Sign Method in python?