Index

A | B | C | D | E | F | G | K | L | M | N | Q | R | S | T | W | Z

A

attention_dropout_rate (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

B

bias_init() (wassersteinwormhole.DefaultConfig.DefaultConfig method)

C

coeff_dec (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

D

E

F

factor (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

G

GS() (in module wassersteinwormhole.utils_OT)

K

kernel_init() (wassersteinwormhole.DefaultConfig.DefaultConfig method)

L

lse_dec (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

lse_enc (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

M

N

num_heads (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

num_layers (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

Q

qkv_dim (wassersteinwormhole.DefaultConfig.DefaultConfig attribute)

R

replace() (wassersteinwormhole.DefaultConfig.DefaultConfig method)

S

S1() (in module wassersteinwormhole.utils_OT)

T

train() (wassersteinwormhole.Wormhole.Wormhole method)

W

W1() (in module wassersteinwormhole.utils_OT)
W2() (in module wassersteinwormhole.utils_OT)
wassersteinwormhole.DefaultConfig
- module

wassersteinwormhole.utils_OT
- module
wassersteinwormhole.Wormhole
- module
Wormhole (class in wassersteinwormhole.Wormhole)

Z

Zeros() (in module wassersteinwormhole.utils_OT)