def test_srelu_share():
from keras.layers.advanced_activations import SReLU
layer_test(SReLU, kwargs={'shared_axes': 1},
input_shape=(2, 3, 4))
微信
公众号
扫码关注公众号