Can Tensorflow Custom activation Layer automatic differentiation and don't need to specify the gradient or weight?

Question

My tensorflow is 2.4.0 , but i am using the code in tensorflow 1 base(sess.run), and I want to change my network activation function to my custom type below , which define in th