Have a question? Connect with the community at the TensorFlow Forum Visit Forum

tfnlp.layers.MatMulWithMargin

This layer computs a dot product matrix given two encoded inputs.

logit_scale The scaling factor of dot products when doing training.
logit_margin The margin value between the positive and negative examples when doing training.

Methods

call

View source

This is where the layer's logic lives.

Note here that call() method in tf.keras is little bit different from keras API. In keras API, you can pass support masking for layers as additional arguments. Whereas tf.keras has compute_mask() method to support masking.

Args
inputs Input tensor, or list/tuple of input tensors.
*args Additional positional arguments. Currently unused.
**kwargs Additional keyword arguments. Currently unused.

Returns
A tensor or list/tuple of tensors.