python - Keras 张量取名

Question

我有一个 keras 张量列表，我为每个张量提供了一个带有 name 参数的名称。
例如，我想将 BatchNormalization layer 命名为 "a":

layer = BatchNormalization(axis=3, name="a")(input)

但是当我尝试检索我为关卡指定的名称时，它看起来像这样:

print(layer.name)
>>> a_2/cond/Merge:0

如何检索我为 layer 对象指定的确切名称(本例中为 “a”)？

Answer 1

让我们将这一行分成两行:

layer = BatchNormalization(axis=3, name="a")
node = layer(input)

现在如果你尝试输出

layer.name 

它将是“a”。

你试图做的是输出这个 batchnorm 层的输出张量的名称，它基本上是一个由此合并操作产生的 tensorflow 张量。

因此，要做到这一点，您必须如上所示对图层执行此操作，或者对操作名称进行一些处理，因为它始终包含图层名称。