Function embedding

python/singa/autograd.py:5715–5722

Produces an embedding operator. Args: Returns: the output Tensor.

(x, w)

def embedding(x, w):
    """
    Produces an embedding operator.
    Args:
    Returns:
        the output Tensor.
    """
    return Embedding()(x, w)[0]


class Erf(Operator):