mttf 1.2.24__py3-none-any.whl → 1.2.26__py3-none-any.whl

mt/keras_src/layers_src/__init__.py

@@ -13,6 +13,7 @@ from .simple_mha import *
 from .image_sizing import *
 from .counter import Counter
 from .normed_conv2d import NormedConv2D
+from .soft_bend import SoftBend
 from .utils import *
 
 
@@ -36,4 +37,5 @@ __api__ = [
     "Counter",
     "conv2d",
     "dense2d",
+    "SoftBend",
 ]

mt/keras_src/layers_src/soft_bend.py

@@ -0,0 +1,27 @@
+from .. import layers
+
+
+class SoftBend(layers.Layer):
+    """Soft bend activation layer.
+
+    Function: `|x|^alpha * tanh(x)`, bending the linear activation a bit.
+
+    If alpha is less than 1, it acts as a soft squash.
+    If alpha is greater than 1, it acts as a soft explode.
+    If alpha is 1, it acts as the linear activation function.
+    """
+
+    def __init__(self, alpha: float = 0.5):
+        self.alpha = alpha
+
+    def call(self, x):
+        from tensorflow.math import pow, abs, tanh
+
+        return pow(abs(x), self.alpha) * tanh(x)
+
+    call.__doc__ = layers.Layer.call.__doc__
+
+    def compute_output_shape(self, input_shape):
+        return input_shape
+
+    compute_output_shape.__doc__ = layers.Layer.compute_output_shape.__doc__

mt/tf/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 2
-PATCH_VERSION = 24
+PATCH_VERSION = 26
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mttf-1.2.24.dist-info → mttf-1.2.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mttf
-Version: 1.2.24
+Version: 1.2.26
 Summary: A package to detect and monkey-patch TensorFlow and Keras, for Minh-Tri Pham
 Home-page: https://github.com/inteplus/mttf
 Author: ['Minh-Tri Pham']

{mttf-1.2.24.dist-info → mttf-1.2.26.dist-info}/RECORD

@@ -7,34 +7,35 @@ mt/keras_src/applications_src/mobilenet_v3_split.py,sha256=EufzlbM2B5D_whpsDmfJF
 mt/keras_src/applications_src/mobilevit.py,sha256=WSwTTT_VTPkH03XmRB_tFS1IgZf__HcVHwshjidSKaM,8934
 mt/keras_src/constraints_src/__init__.py,sha256=XOHrlZbWjmQszPm9TGQFSmKthbcgBvb1jXUlgZbtyMM,253
 mt/keras_src/constraints_src/center_around.py,sha256=NIPckqjbrOfGCT3Fv1O3hET4sHgE7kBWIMt5AG03Lmk,543
-mt/keras_src/layers_src/__init__.py,sha256=EnDoayl_SVb5cYyISiGbe0v5IPoklE4BgyJk-ft8SrI,746
+mt/keras_src/layers_src/__init__.py,sha256=mqFyOSZiwGzvhorE8icMe2VyuX42YOvDLpragANXbKo,794
 mt/keras_src/layers_src/counter.py,sha256=pFN50ZOpV4lvkKmKsK5hxfvAF0Jj46TtGqZOtND5eSc,852
 mt/keras_src/layers_src/floor.py,sha256=wPINhDnRYdbJ_eQqwq0pCXiez1sAryvGthQJULeN9yM,508
 mt/keras_src/layers_src/identical.py,sha256=sxY0E8BTk2F_fflGtcrrSv9d-xf47SUdAJezdNM5HkM,357
 mt/keras_src/layers_src/image_sizing.py,sha256=JcxBJ_JqFsrZAtlJrRy9Yi9Q1hslAYopbipo9cbGF60,58525
 mt/keras_src/layers_src/normed_conv2d.py,sha256=IBuAxwjLDxb02WDPFteEYSv4IsF9PYIcQDLUrc6I8lU,10676
 mt/keras_src/layers_src/simple_mha.py,sha256=jnp8e13Xk10BvR4wverOVI14T3QEnIXaMRKV0RYIFh8,19233
+mt/keras_src/layers_src/soft_bend.py,sha256=arOgrFk3Udv1K8riU2y8OfSBTnvSJwt0_tygCTLuwso,747
 mt/keras_src/layers_src/utils.py,sha256=cvOikMkSSCJ9Si-eIfp_hD86AQ-PxbI5xtIfp9bcW8E,2806
 mt/keras_src/layers_src/var_regularizer.py,sha256=yJwNCtetTT-a4Y38PXEUCyl3OQQzIxT9ybj9jP1r25A,1104
 mt/tf/__init__.py,sha256=M8xiJNdrAUJZgiZTOQOdfkehjO-CYzGpoxh5HVGBkms,338
 mt/tf/init.py,sha256=bcm0t5tstxTkCBOiMX1SJxhKOz0MAZf-MYp2Mk0Gtas,502
 mt/tf/mttf_version.py,sha256=ha53i-H9pE-crufFttUECgXHwPvam07zMKzApUts1Gs,206
 mt/tf/utils.py,sha256=wau2vhPoPHu2cDxlc2lc9fxrndOXPdq2DNG4em5OOMI,1025
-mt/tf/version.py,sha256=6FaoVwSYTyaipLFHUyDW3EHsEl4mM_bdUbuwZOMtaHg,207
+mt/tf/version.py,sha256=JFDL5ioQIIDH85km6zu3xIDAMjI3olVWaNchkZAxCqk,207
 mt/tf/keras_applications/__init__.py,sha256=m-A1rHGGLQgHX9690ENWXZkrU0vqfsJkZXcjIG3CLM0,142
 mt/tf/keras_layers/__init__.py,sha256=NsuFD-kSuy6cVV3Kl7ab95tw4g7x4Igv3cF-Ky3VuCo,124
 mt/tfc/__init__.py,sha256=pcxgcmUSXJCFZal5Pxz7hC1P8fhP1dB2egZzKGdrseQ,12118
 mt/tfg/__init__.py,sha256=6Ly2QImAyQTsg_ZszuAuK_L2n56v89Cix9yYmMVk0CM,304
 mt/tfp/__init__.py,sha256=AQkGCkmDRwswEt3qoOSpxe-fZekx78sHHBs2ZVz33gc,383
 mt/tfp/real_nvp.py,sha256=U9EmkXGqFcvtS2yeh5_RgbKlVKKlGFGklAb7Voyazz4,4440
-mttf-1.2.24.data/scripts/dmt_pipi.sh,sha256=NNsj4P332unHMqU4mAFjU9PQvxQ8TK5XQ42LC29IZY8,510
-mttf-1.2.24.data/scripts/dmt_twineu.sh,sha256=KZhcYwuCW0c36tWcOgCe7uxJmS08rz-J6YNY76Exy4M,193
-mttf-1.2.24.data/scripts/pipi.sh,sha256=kdo96bdaKq2QIa52Z4XFSiGPcbDm09SAU9cju6I2Lxo,289
-mttf-1.2.24.data/scripts/wml_nexus.py,sha256=47P9PQMgb9w_-T0olC-dr3s60mKaQup-RWOuNi5mvJg,1192
-mttf-1.2.24.data/scripts/wml_pipi.sh,sha256=CuidIcbuxyXSBNQqYRhCcSC8QbBaSGnQX0KAIFaIvKA,499
-mttf-1.2.24.data/scripts/wml_twineu.sh,sha256=av1JLN765oOWC5LPkv2eSWIVof26y60601tMGkuYdb8,180
-mttf-1.2.24.dist-info/licenses/LICENSE,sha256=e_JtcszdGZ2ZGfjcymTGrcxFj_9XPicZOVtnsrPvruk,1070
-mttf-1.2.24.dist-info/METADATA,sha256=G0n_8yIDPY33sfF41lqDz7LjBqmxGRRR0Er7YN4rl2M,568
-mttf-1.2.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mttf-1.2.24.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mttf-1.2.24.dist-info/RECORD,,
+mttf-1.2.26.data/scripts/dmt_pipi.sh,sha256=NNsj4P332unHMqU4mAFjU9PQvxQ8TK5XQ42LC29IZY8,510
+mttf-1.2.26.data/scripts/dmt_twineu.sh,sha256=KZhcYwuCW0c36tWcOgCe7uxJmS08rz-J6YNY76Exy4M,193
+mttf-1.2.26.data/scripts/pipi.sh,sha256=kdo96bdaKq2QIa52Z4XFSiGPcbDm09SAU9cju6I2Lxo,289
+mttf-1.2.26.data/scripts/wml_nexus.py,sha256=47P9PQMgb9w_-T0olC-dr3s60mKaQup-RWOuNi5mvJg,1192
+mttf-1.2.26.data/scripts/wml_pipi.sh,sha256=CuidIcbuxyXSBNQqYRhCcSC8QbBaSGnQX0KAIFaIvKA,499
+mttf-1.2.26.data/scripts/wml_twineu.sh,sha256=av1JLN765oOWC5LPkv2eSWIVof26y60601tMGkuYdb8,180
+mttf-1.2.26.dist-info/licenses/LICENSE,sha256=e_JtcszdGZ2ZGfjcymTGrcxFj_9XPicZOVtnsrPvruk,1070
+mttf-1.2.26.dist-info/METADATA,sha256=pI1BySHACTgRpA0SLPwM9_xmnRtH5lz0q3YIp8qGuqg,568
+mttf-1.2.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mttf-1.2.26.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
+mttf-1.2.26.dist-info/RECORD,,