如何通过保存初始化张量对象？ （无）输入？

Question

说我有一个张量对象h1 = (?, 256)。

如何通过保留'？'将h1初始化为h1 = (?, zeros(256))？ （无）输入？

（即初始化h1[1] = zeros(256)，保留“ None”不变）

我的代码：

h1 = np.zeros((h1.shape[0], h1.shape[1]))    
entryh1[1] = np.zeros(256)

但它不起作用。

Answer 1

您可以使用后端功能zeros_like来实现所需的功能，让我们将其包装在Lambda层上，以便将其传递给其他层：

import keras.backend as K
# ...
gen_zeros = Lambda(lambda x: K.zeros_like(x), name='gen_zeros')
h1_zeros = = gen_zeros(h1) # will give zeros of h1.shape

您可以检查documentation，以获取有关可在自定义图层中使用的后端功能的更多信息。