scSLAT.model.graphmodel.WDiscriminator

class scSLAT.model.graphmodel.WDiscriminator(hidden_size, hidden_size2=512)

Bases: Module

WGAN Discriminator

Parameters:

• hidden_size (int) – input dim

• hidden_size2 (typing.Optional[int]) – hidden dim

Methods

forward

Defines the computation performed at every call.

Attributes