View source on GitHub
|
Sampling probability correction.
tfrs.layers.loss.SamplingProbablityCorrection(
trainable=True, name=None, dtype=None, dynamic=False, **kwargs
)
Methods
call
call(
inputs, **kwargs
)
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.
| Arguments | |
|---|---|
inputs
|
Input tensor, or list/tuple of input tensors. |
**kwargs
|
Additional keyword arguments. Currently unused. |
| Returns | |
|---|---|
| A tensor or list/tuple of tensors. |