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


Static Public Attributes


Detailed Description

Definition at line 23 of file

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