Hello Dear. I am trying to use this layer by converting it to tf.keras but the weight can be setted. Any help/