parameters: (module)
return_statement: return type(module) in [nn.Linear, nn.Conv1d, nn.Conv2d, nn.Conv3d]
docstring: Checks if the torch.nn node supports equalization.
docstring_summary: Checks if the torch.nn node supports equalization.
docstring_tokens: Checks if the torch . nn node supports equalization .
function: def nn_module_supports_equalization(module) -> bool:
      """ Checks if the torch.nn node supports equalization. """
      return type(module) in [nn.Linear, nn.Conv1d, nn.Conv2d, nn.Conv3d]
function_tokens: def nn_module_supports_equalization ( module ) -> bool : return type ( module ) in [ nn . Linear , nn . Conv1d , nn . Conv2d , nn . Conv3d ]