public SDVariable embeddingLookup(SDVariable x, SDVariable[] indices,
      PartitionMode PartitionMode) {
    SDValidation.validateNumerical("EmbeddingLookup", "x", x);
    SDValidation.validateInteger("EmbeddingLookup", "indices", indices);
    Preconditions.checkArgument(indices.length >= 1, "indices has incorrect size/length. Expected: indices.length >= 1, got %s", indices.length);