torch.nn.modules.adaptive.AdaptiveLogSoftmaxWithLoss Class Reference

Inheritance diagram for torch.nn.modules.adaptive.AdaptiveLogSoftmaxWithLoss:

Public Member Functions
def	__init__ (self, in_features, n_classes, cutoffs, div_value=4., head_bias=False)
def	reset_parameters (self)
def	forward (self, input, target)
def	log_prob (self, input)
def	predict (self, input)

Public Attributes
	in_features
	n_classes
	cutoffs
	div_value
	head_bias
	shortlist_size
	n_clusters
	head_size
	head
	tail

Detailed Description

Definition at line 15 of file adaptive.py.

---

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

• torch/nn/modules/adaptive.py