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:

• torch/nn/modules/activation.py