Thanks~ but it is still so many codes, a lambda layer like the one used in keras would be very helpful.