oneflow.one_embedding.make_uniform_initializer

oneflow.one_embedding.make_uniform_initializer(low=0.0, high=1.0)

make uniform initializer param of make_table_options

Parameters
  • low (float) – A python scalar. Lower bound of the range of random values to generate.

  • high (float) – A python scalar. Upper bound of the range of random values to generate.

Returns

initializer param of make_table_options

Return type

dict

For example:

>>> import oneflow as flow
>>> initializer = flow.one_embedding.make_uniform_initializer(low=-scale, high=scale)
>>> # pass the initializer to flow.one_embedding.make_table_options
>>> # ...