马春杰杰这是最好不要使用的情况之一，最好自己从模型继承并执行所需的操作。
例如nn.Sequential
|
1 2 3 4 |
class MyModel(nn.Module): def forward(self, input): return input ** 2 + 1 model = MyModel() |
但是，如果你想要一个等效的层，你可以很容易在pytorch中写它Lambda
|
1 2 3 4 5 6 |
class LambdaLayer(nn.Module): def __init__(self, lambd): super(LambdaLayer, self).__init__() self.lambd = lambd def forward(self, x): return self.lambd(x) |
现在你可以用它， 就像你在Keras中一样：
|
1 |
model.add(Lambda(lambda x: x ** 2)) |
本文最后更新于2020年12月3日，已超过 1 年没有更新，如果文章内容或图片资源失效，请留言反馈，我们会及时处理，谢谢！
123 2个月前 (03-20)说：
查克拉 2个月前 (03-19)说：
mumujun12345 2个月前 (03-18)说：
感谢分享啊 2个月前 (03-16)说：
langjitianya 3个月前 (03-04)说：
骏 4个月前 (01-28)说：
爱上浅井 5个月前 (12-30)说：
123 5个月前 (12-25)说：
