def apply(self, is_train, x, mask=None):
mask = tf.sequence_mask(mask, tf.shape(x)[1])
output = weight_layers(1, x, mask, self.l2_coef, do_layer_norm=self.layer_norm,
use_top_only=self.top_layer_only)["weighted_ops"][0]
return output
评论列表
文章目录