> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/sdgm/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/sdgm/_mcp/server.

# output_embedding_dim

#### `output_embedding_dim: <list[int]>` (Optional)

## Description

A list of output embedding dimensions for link prediction models for AutoML to explore.

`output_embedding_dim` must satisfy `0 < output_embedding <= 256`.

### Supported Task Types

* Link Prediction

### Default Values

| [run\_mode](/run-mode) | Default Value |
| ---------------------- | ------------- |
| FAST                   | `[32]`        |
| NORMAL                 | `[32]`        |
| BEST                   | `[32]`        |