Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
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: