6bi |SrSSKJs Jr SSKJr SSKJr SSK J r SSK J r Sr \ "S5"S S \55rg) zSoftmax activation layer.N)backend)Layer)tf_utils) keras_exportcfU[R:Xa[RRS- $g)zLarge 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. g@ge)tffloat16min) tensor_types `/home/james-whalen/.local/lib/python3.13/site-packages/tf_keras/src/layers/activation/softmax.py_large_compatible_negativer s&bjj zz~~## zkeras.layers.Softmaxcj^\rSrSrSrSU4SjjrS SjrU4Sjr\RS5r Sr U=r $) Softmax0aSoftmax 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`. The mask specifies 1 to keep and 0 to mask. Defaults to `None`. Returns: Softmaxed output with the same shape as `inputs`. c @>[TU]"S0UD6 SUlXlg)NT)super__init__supports_maskingaxis)selfrkwargs __class__s r rSoftmax.__init__Ss  "6" $ rc Ub=S[R"X!R5- [UR5-nX- n[ UR [ [45(ap[UR 5S:a6[R"U[R"XR SS9- 5$[R"XR SS9$[R"XR S9$)Ng?T)rkeepdimsr)r) rcastdtyper isinstancertuplelistlenexpreduce_logsumexprsoftmax)rinputsmaskadders r call Softmax.callXs  277466*6<<8E OF dii% / /499~!vv))&yy4PQ vIIaLAAvII66rc>SUR0n[TU] 5n[[ UR 55[ UR 55-5$)Nr)rr get_configdictr#items)rconfig base_configrs r r.Softmax.get_confignsG$))$g(* D**,-V\\^0DDEErcU$Nr)r input_shapes r compute_output_shapeSoftmax.compute_output_shapessr)rr)r5) __name__ __module__ __qualname____firstlineno____doc__rr+r.rshape_type_conversionr7__static_attributes__ __classcell__)rs@r rr0s3B 7,F ##$rr)r>tensorflow.compat.v2compatv2r tf_keras.srcrtf_keras.src.engine.base_layerrtf_keras.src.utilsr tensorflow.python.util.tf_exportrr rrrr rIsI "! 0':($%DeD&Dr