oneflow.one_embedding.make_normal_initializer

oneflow.one_embedding.make_normal_initializer(mean=0.0, std=1.0)

make normal initializer param of make_table_options

Parameters
  • mean (float) – A python scalar. Mean of the random values to generate.

  • std (float) – A python scalar. Standard deviation of the 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_normal_initializer(mean=0, std=0.01)
>>> # pass the initializer to flow.one_embedding.make_table_options
>>> # ...