deepr.layers.Embedding
- class deepr.layers.Embedding(variable_name, shape, trainable=True, initializer=None, num_shards=None, reuse=None, partition_strategy='div', **kwargs)[source]
Partitioned Embedding Layer
- __init__(variable_name, shape, trainable=True, initializer=None, num_shards=None, reuse=None, partition_strategy='div', **kwargs)[source]
Methods
__init__
(variable_name, shape[, trainable, ...])forward
(tensors[, mode])Forward method of the layer
forward_as_dict
(tensors[, mode])Forward method on a dictionary of Tensors.