mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-07-28 11:24:50 +08:00
8 lines
176 B
Python
8 lines
176 B
Python
import inspect
|
|
|
|
import torch.nn as nn
|
|
|
|
def get_module_forward_input_names(module: nn.Module):
|
|
params = inspect.signature(module.forward).parameters
|
|
return list(params)
|