CDDecoder

Module Contents

class CDDecoder(num_layers, vocab_size, bias_p, bias_topk)

Bases: torch.nn.Module

num_layers
bias_p
bias_topk
decoder_bn_list
bias_vectors
forward(input_bow, theta_list, beta_list)