qumphy.models.utils.transformer_encdec module

class qumphy.models.utils.transformer_encdec.ConvLayer(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x)[source]
class qumphy.models.utils.transformer_encdec.Decoder(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x, cross, x_mask=None, cross_mask=None, tau=None, delta=None)[source]
class qumphy.models.utils.transformer_encdec.DecoderLayer(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x, cross, x_mask=None, cross_mask=None, tau=None, delta=None)[source]
class qumphy.models.utils.transformer_encdec.Encoder(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x, attn_mask=None, tau=None, delta=None)[source]
class qumphy.models.utils.transformer_encdec.EncoderLayer(*args: Any, **kwargs: Any)[source]

Bases: Module

forward(x, attn_mask=None, tau=None, delta=None)[source]