torch::nn::EmbeddingOptions Struct Reference

Options for the Embedding module. More...

#include <embedding.h>

Public Member Functions
	EmbeddingOptions (int64_t count, int64_t dimension)
	TORCH_ARG (int64_t, count)
	The number of embeddings (number of rows in the table).
	TORCH_ARG (int64_t, dimension)
	The size of each embedding vector (number of columns in the table).

Detailed Description

Options for the Embedding module.

Definition at line 14 of file embedding.h.

---

The documentation for this struct was generated from the following files:

• torch/csrc/api/include/torch/nn/modules/embedding.h
• torch/csrc/api/src/nn/modules/embedding.cpp