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

Public Member Functions

def __init__ (self, dim=None)
 
def __setstate__ (self, state)
 
def forward (self, input)
 

Public Attributes

 dim
 

Detailed Description

Definition at line 830 of file activation.py.


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