cascade.nn.AttnPool

class cascade.nn.AttnPool(emb_dim)[source]

Bases: Module

Attention-based pooling layer to combine multiple intervention embeddings

Parameters:: emb_dim (int) – Embedding dimensionality

Methods