ProbabilityEntry¶

class paddle.distributed. ProbabilityEntry ( probability ) [source]

Examples

import paddle

sparse_feature_dim = 1024
embedding_size = 64

entry = paddle.distributed.ProbabilityEntry(0.1)

input = paddle.static.data(name='ins', shape=[1], dtype='int64')

emb = paddle.static.nn.sparse_embedding((
    input=input,
    size=[sparse_feature_dim, embedding_size],
    is_test=False,
    entry=entry,
    param_attr=paddle.ParamAttr(name="SparseFeatFactors",
                               initializer=paddle.nn.initializer.Uniform()))