【发布时间】:2020-02-25 07:47:15
【问题描述】:
我在这个简单的层中有以下错误:
class MyLayer(Layer):
def __init__(self):
super(MyLayer, self).__init__()
def build(self):
# Create a trainable weight variable for this layer.
self.kernel = self.add_weight(name='kernel',
shape=(1)
trainable=True)
super(MyLayer, self).build()
def call(self, x):
return x/self.kernel
当我把它用作:
m = MyLayer()
t = m (input)
错误:build() 采用一个位置参数,但给出了两个。
【问题讨论】:
-
@giser_yugang 是同一个例子..
-
Keras 中的每一层都需要一个
input_shape参数,而您的build()方法中不存在该参数。 -
@ShubhamPanchal 我说的是'int object is not iterable'错误
-
您需要像
for i in range( your_int )一样进行迭代。在您的问题中包含该代码。\
标签: python tensorflow keras keras-layer tf.keras