There was an error while loading. Please reload this page.
1 parent b39a2f2 commit 2ef7331Copy full SHA for 2ef7331
torch/nn/modules/sparse.py
@@ -18,7 +18,7 @@ class Embedding(Module):
18
max_norm (float, optional): If given, will renormalize the embeddings to always have a norm lesser than this
19
norm_type (float, optional): The p of the p-norm to compute for the max_norm option
20
scale_grad_by_freq (boolean, optional): if given, this will scale gradients by the frequency of
21
- the words in the dictionary.
+ the words in the mini-batch.
22
23
Attributes:
24
weight (Tensor): the learnable weights of the module of shape (num_embeddings, embedding_dim)
0 commit comments