tf.tpu.experimental.embedding.RowIdInitializer

An initializer that initializes the table with vocabulary ids.

Compat aliases for migration

tf.compat.v1.tpu.experimental.embedding.RowIdInitializer

tf.tpu.experimental.embedding.RowIdInitializer(
    offset: int = 0
)

Methods

__call__(
    shape: Union[Sequence[int], tf.TensorShape],
    dtype: tf.dtypes.DType
) -> core.Tensor

Call self as a function.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/tpu/experimental/embedding/RowIdInitializer