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.