Get arguments for torch.nn.functional methods

torch.compile FX (Functional Transformations)
1

Hi,

Is there any way to extract the argument names / order for methods in torch.nn.functional? If I try using the inspect module I get a TypeError('unsupported callable'), e.g.

inspect.getargspec(torch.nn.functional.conv2d)  # TypeError: unsupported callable
inspect.signature(torch.nn.functional.conv2d)  # ValueError: no signature found for builtin <built-in method conv2d of type object at 0x7f7a3ce19780>

I’m trying to save the kernel size, stride, etc. without having to hard-code which arguments they come from.

Thanks in advance. -Collin