Infinity Attack Function
Details about the Infinity attack method.
def inf(model_size, device):
return torch.ones(attack.model_size, device=attack.device) * float('inf')
Inputs
model_size: int - Size of the model.device: str - Device to use for the attack.
Outputs
torch.Tensor - The processed Tensor filled with infinity values.