代码之家 › 专栏 › 技术社区 › aerin

pytorch在哪里实现了嵌入“max\u norm”?

pytorch

aerin · 技术社区 · 6 年前

max_norm (float, optional) â If given, will renormalize the embedding vectors to have a norm lesser than this before extracting.

1) 在我的模型中,我使用这个嵌入类作为参数,而不仅仅是作为输入(模型学习嵌入)。在这种情况下,我假设每次更新发生时,嵌入都会被重新规范化,而不仅仅是在初始化时。我的理解正确吗?

1 回复 | 直到 6 年前

Umang Gupta 6 年前

如果你看到 forward 嵌入类中的函数 here ,有一个参考 torch.nn.functional.embedding here 这意味着它是一个cpp实现。在pytorch repo上的一些github搜索指向这个文件( 1 , 2

对1的回答是肯定的。答案2在上面。

推荐文章

Ze0ruso · 做模型。eval()在代码中的位置有什么关系?

2 年前

rkraaijveld · sklearn的Coef。线性回归为无

2 年前

Ellis Thompson · 填充一个numpym数组以满足所需的大小

2 年前

Adam · 如何在多个模型上并行运行PyTorch推理?

2 年前

quinlan · 是dgl。nn。皮托克。GraphConv聚合边缘信息?

2 年前

Ofek Nourian · 完全连通的简单自回归模型预测

2 年前

MrThiele1708 · 运行时错误:mat1和mat2形状不能相乘(5400x64和5400x64)

2 年前

Moodhi · 计算GEC的F分数

2 年前

john wang · matlab使用可调用的python函数,但需要输入

2 年前

odbhut.shei.chhele · 如何解决错误:RuntimeError:收到0项ancdata

2 年前