Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
torch.nn.modules.module.Module Class Reference
Inheritance diagram for torch.nn.modules.module.Module:

Public Member Functions

def __init__ (self)
 
def forward (self, input)
 
def register_buffer (self, name, tensor)
 
def register_parameter (self, name, param)
 
def add_module (self, name, module)
 
def apply (self, fn)
 
def cuda (self, device=None)
 
def cpu (self)
 
def type (self, dst_type)
 
def float (self)
 
def double (self)
 
def half (self)
 
def to (self, args, kwargs)
 
def register_backward_hook (self, hook)
 
def register_forward_pre_hook (self, hook)
 
def register_forward_hook (self, hook)
 
def __call__ (self, input, kwargs)
 
def __setstate__ (self, state)
 
def __getattr__ (self, name)
 
def __setattr__ (self, name, value)
 
def __delattr__ (self, name)
 
def state_dict (self, destination=None, prefix='', keep_vars=False)
 
def load_state_dict (self, state_dict, strict=True)
 
def parameters (self, recurse=True)
 
def named_parameters (self, prefix='', recurse=True)
 
def buffers (self, recurse=True)
 
def named_buffers (self, prefix='', recurse=True)
 
def children (self)
 
def named_children (self)
 
def modules (self)
 
def named_modules (self, memo=None, prefix='')
 
def train (self, mode=True)
 
def eval (self)
 
def zero_grad (self)
 
def share_memory (self)
 
def extra_repr (self)
 
def __repr__ (self)
 
def __dir__ (self)
 

Public Attributes

 training
 

Static Public Attributes

 dump_patches
 

Detailed Description

Definition at line 23 of file module.py.


The documentation for this class was generated from the following file: