~WhdZddlmcmZddlmZddlmZddl m Z ddl m Z dZ e dGd d eZdS) zSoftmax activation layer.N)backend)Layer)tf_utils) keras_exportcH|tjkrtjjSdS)aLarge negative number as Tensor. This function is necessary because the standard value for epsilon in this module (-1e9) cannot be represented using tf.float16 Args: tensor_type: a dtype to determine the type. Returns: a large negative number. ge)tffloat16min) tensor_types e/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/keras/layers/activation/softmax.py_large_compatible_negativer s bj  z~ 4zkeras.layers.SoftmaxcVeZdZdZdfd Zd dZfdZejdZ xZ S) SoftmaxaSoftmax activation function. Example without mask: >>> inp = np.asarray([1., 2., 1.]) >>> layer = tf.keras.layers.Softmax() >>> layer(inp).numpy() array([0.21194157, 0.5761169 , 0.21194157], dtype=float32) >>> mask = np.asarray([True, False, True], dtype=bool) >>> layer(inp, mask).numpy() array([0.5, 0. , 0.5], dtype=float32) Input shape: Arbitrary. Use the keyword argument `input_shape` (tuple of integers, does not include the samples axis) when using this layer as the first layer in a model. Output shape: Same shape as the input. Args: axis: Integer, or list of Integers, axis along which the softmax normalization is applied. Call arguments: inputs: The inputs, or logits to the softmax layer. mask: A boolean mask of the same shape as `inputs`. Defaults to `None`. The mask specifies 1 to keep and 0 to mask. Returns: softmaxed output with the same shape as `inputs`. c Vtjdi|d|_||_dS)NT)super__init__supports_maskingaxis)selfrkwargs __class__s r rzSoftmax.__init__Os3""6""" $ rNc|7dtj||jz t|jz}||z }t |jt tfrjt|jdkr1tj |tj ||jdz Stj ||jdStj ||jS)Ng?T)rkeepdimsr)r) rcastdtyper isinstancertuplelistlenexpreduce_logsumexprsoftmax)rinputsmaskadders r callz Softmax.callTs  274666*6<88E eOF di% / / B49~~!!v)&ty4PPPQ vDIaLAAAAvDI6666rcd|ji}t}tt |t |zS)Nr)rr get_configdictr"items)rconfig base_configrs r r,zSoftmax.get_configjs[$)$gg((** D**,,--V\\^^0D0DDEEErc|SNr)r input_shapes r compute_output_shapezSoftmax.compute_output_shapeosr)rr2) __name__ __module__ __qualname____doc__rr*r,rshape_type_conversionr4 __classcell__)rs@r rr-s@ 7777,FFFFF #$#rr)r8tensorflow.compat.v2compatv2rkerasrkeras.engine.base_layerr keras.utilsr tensorflow.python.util.tf_exportrr rrrr rBs "!!!!!!!!)))))) :99999"$%%CCCCCeCC&%CCCr