CDDecoder ========= .. py:module:: topmost.models.hierarchical.TraCo.CDDecoder Module Contents --------------- .. autoapisummary:: topmost.models.hierarchical.TraCo.CDDecoder.CDDecoder .. py:class:: CDDecoder(num_layers, vocab_size, bias_p, bias_topk) Bases: :py:obj:`torch.nn.Module` .. py:attribute:: num_layers .. py:attribute:: bias_p .. py:attribute:: bias_topk .. py:attribute:: decoder_bn_list .. py:attribute:: bias_vectors .. py:method:: forward(input_bow, theta_list, beta_list)