torch.mutation

user_input_mutation

Note

Tags: torch.mutation

Support Level: NOT_SUPPORTED_YET

Original source code:

import torch



class UserInputMutation(torch.nn.Module):
    """
    Can't directly mutate user input in forward
    """

    def forward(self, x):
        x.mul_(2)
        return x.cos()

Result:

RuntimeError: