代码之家 › 专栏 › 技术社区 › mrgloom

喀拉斯:有没有可能给lamda层起个名字?

deep-learning keras python

0

mrgloom · 技术社区 · 7 年前

我的模型定义中有以下代码

# (None, 6) -> (None, 8)
def affine_to_projective(x):
    y = tf.pad(x,
               paddings=tf.constant([[0, 0], [0, 2]]),
               mode='CONSTANT',
               constant_values=0
               )
    return y

projective_transform = Lambda(affine_to_projective)(affine_transform)

在模型摘要中,它被视为

lambda_1 (Lambda)               (None, 8)            0           affine_transform[0][0]

是否可以给这个层一个名称而不是 lambda_1 ?

1 回复 | 直到 7 年前

1

Daniel Möller 7 年前

projective_transform = Lambda(affine_to_projective, name='lambda_name')(affine_transform)