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

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

  •  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
  •  1
  •   Daniel Möller    7 年前
    projective_transform = Lambda(affine_to_projective, name='lambda_name')(affine_transform)