export.OnnxExpertWrapper Class Reference

Inheritance diagram for export.OnnxExpertWrapper:

Collaboration diagram for export.OnnxExpertWrapper:

Public Member Functions
	__init__ (self, nn.Module expert)
	forward (self, torch.Tensor features)

Public Attributes
	expert = expert

Constructor & Destructor Documentation

__init__()

export.OnnxExpertWrapper.__init__	(		self,
		nn.Module	expert )

Here is the call graph for this function:

Here is the caller graph for this function:

Member Function Documentation

forward()

export.OnnxExpertWrapper.forward	(		self,
		torch.Tensor	features )

Member Data Documentation

expert

export.OnnxExpertWrapper.expert = expert

---

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

• training/export.py