问火炬传递中的暹罗神经网络
EN

Stack Overflow用户

提问于 2018-12-16 15:57:10

回答 1查看 3.4K关注 0票数 5

如何在PyTorch中实现暹罗神经网络？

什么是暹罗神经网络？一个暹罗神经网络由两个相同的神经网络组成，每个神经网络只接受一个输入。相同的意思是两个神经网络具有完全相同的结构和相同的权重。

python

pytorch

回答 1

Stack Overflow用户

回答已采纳

发布于 2018-12-16 15:57:10

在PyTorch中实现siamese神经网络就像在不同的输入上两次调用网络函数一样简单。

mynet = torch.nn.Sequential(
        nn.Linear(10, 512),
        nn.ReLU(),
        nn.Linear(512, 2))
...
output1 = mynet(input1)
output2 = mynet(input2)
...
loss.backward()

当调用loss.backwad()时，PyTorch将自动对来自mynet的两个调用的梯度进行求和。

您可以找到一个完整的示例这里。

票数 8

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/53803889

复制

相似问题

问火炬传递中的暹罗神经网络
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问火炬传递中的暹罗神经网络EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问火炬传递中的暹罗神经网络
EN