wassersteinwormhole

• wassersteinwormhole.Wormhole module
  • Wormhole
    • Wormhole.decode()
    • Wormhole.encode()
    • Wormhole.train()
• wassersteinwormhole.DefaultConfig module
  • DefaultConfig
    • DefaultConfig.attention_dropout_rate
    • DefaultConfig.bias_init()
    • DefaultConfig.coeff_dec
    • DefaultConfig.dist_func_dec
    • DefaultConfig.dist_func_enc
    • DefaultConfig.dtype
    • DefaultConfig.emb_dim
    • DefaultConfig.eps_dec
    • DefaultConfig.eps_enc
    • DefaultConfig.factor
    • DefaultConfig.kernel_init()
    • DefaultConfig.lse_dec
    • DefaultConfig.lse_enc
    • DefaultConfig.mlp_dim
    • DefaultConfig.num_heads
    • DefaultConfig.num_layers
    • DefaultConfig.qkv_dim
    • DefaultConfig.replace()
    • DefaultConfig.scale
• wassersteinwormhole.utils_OT module
  • GS()
  • GS_scale()
  • GW()
  • S1()
  • S2()
  • W1()
  • W2()
  • Zeros()