class_docstr: Returns a layer that computes custom logits for BERT squad model.
class_docstr_tok: ["Returns","a","layer","that","computes","custom","logits","for","BERT","squad","model","."]
code: def __init__(self, initializer=None, float_type=tf.float32, **kwargs):
      super(BertSquadLogitsLayer, self).__init__(**kwargs)
      self.initializer = initializer
      self.float_type = float_type
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tensorflow_hub
  from utils import tf_utils
  from utils.modeling import losses
  from utils.modeling import networks
  from utils.modeling.networks import bert_classifier
  from utils.modeling.networks import bert_pretrainer
  from utils.modeling.networks import bert_span_labeler
label_desc: Use image node_id 1 to create a new BertSquadLogitsLayer object from inherited base classes: tf.keras.layers with example: obj = BertSquadLogitsLayer(initializer, float_type)