Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings


View source on GitHub

Solves the maximization problem weights^T*x with the constraint norm(x)=1.


This op solves a batch of maximization problems at one time. The first axis of weights is assumed to be the batch dimension, and each "row" is treated as an independent maximization problem.

This op is mainly used to generate adversarial examples (e.g., FGSM proposed by Goodfellow et al.). Specifically, the weights are gradients, and x is the adversarial perturbation. The desired perturbation is the one causing the largest loss increase. In this op, the loss increase is approximated by the dot product between the gradient and the perturbation, as in the first-order Taylor approximation of the loss function.


  • weights: tensor representing a batch of weights to define the maximization objective.
  • norm_type: one of nsl.configs.NormType, the type of vector norm.


A tensor representing a batch of adversarial perturbations as the solution to the maximization problems. The returned tensor has the same shape and type as the input weights.