Bridges

[source]

NoOpBridge

polyaxon.bridges.noop_bridge.NoOpBridge(mode, state_size=None, name='NoOpBridge')

A bridge that passes the encoder output to the decoder.

  • Args:
    • mode: str. Specifies if this training, evaluation or prediction. See Modes.
    • name: str. The name of this bridge, used for creating the scope.
    • state_size: int. The bridge state size. Default None, it will be inferred directly from the incoming tensor.

[source]

LatentBridge

polyaxon.bridges.latent_bridge.LatentBridge(mode, latent_dim=1, state_size=None, mean=0.0, stddev=1.0, name='LatentBridge')

A bridge that create a latent space based on the encoder output.

This bridge could be used by VAE.

  • Args:

    • latent_dim: int. The latent dimension to use.
    • mode: str. Specifies if this training, evaluation or prediction. See Modes.
    • name: str. The name of this subgraph, used for creating the scope.
  • Attributes:

    • z_mean: Tensor. The latent distribution mean.
    • z_log_sigma: Tensor. The latent distribution log variance.