Searched defs:hard_distr_embed (Results 1 – 1 of 1) sorted by relevance
159 hard_distr_embed = tf.math.sigmoid(quant_embed[:, 4 * nsymbols : ]).numpy() variable181 hard_distr_embed = np.broadcast_to(hard_distr_embed, (sym_batch.shape[0], sym_batch.shape[1], 2*sym… variable