Tensorflow 编写自己的层
我对凯拉斯很陌生。 我正在尝试(使用函数API)在Keras中编写一个自定义层,就像在Keras.io中一样Tensorflow 编写自己的层,tensorflow,keras,keras-layer,Tensorflow,Keras,Keras Layer,我对凯拉斯很陌生。 我正在尝试(使用函数API)在Keras中编写一个自定义层,就像在Keras.io中一样 class MyLayer(Layer): def __init__(self, output_dim, **kwargs): self.output_dim = output_dim super(MyLayer, self).__init__(**kwargs) def build(self, input_shape):
class MyLayer(Layer):
def __init__(self, output_dim, **kwargs):
self.output_dim = output_dim
super(MyLayer, self).__init__(**kwargs)
def build(self, input_shape):
# Create a trainable weight variable for this layer.
self.kernel = self.add_weight(shape=(input_shape[1], self.output_dim),
initializer='uniform',
trainable=True)
super(MyLayer, self).build(input_shape) # Be sure to call this somewhere!
def call(self, x):
return K.dot(x, self.kernel)
def compute_output_shape(self, input_shape):
return (input_shape[0], self.output_dim)
作为:
sourcedistribution = Mylayer(dropout2_target)
错误回溯如下所示:
File "/root/PycharmProjects/chNET/modelBuild.py", line 77, in
model = create_network([100, 100, 3])
File "/root/PycharmProjects/chNET/modelBuild.py", line 53, in create_network
sourcedistribution = Mylayer(dropout2_source)
NameError: global name 'Mylayer' is not defined
请帮助理解此错误。我认为唯一的问题是您已经定义了
MyLayer
,但您称之为MyLayer