| Server IP : 217.160.0.135 / Your IP : 216.73.217.25 Web Server : Apache System : Linux www 6.18.52-i1-ampere #1203 SMP Mon Sep 14 18:29:59 CEST 2026 aarch64 User : sws1074145052 ( 1074145052) PHP Version : 8.3.32 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /usr/lib/python3/dist-packages/scipy/special/__pycache__/ |
Upload File : |
�
d�cK! � �: � d dl Zd dlmZ g d�Zdd�Zd d�Zd d�ZdS )
� N)�_asarray_validated)� logsumexp�softmax� log_softmaxFc � � t | d�� � } |�It j | |� � \ } }t j |dk � � r| dz } t j | |dk < t j | |d�� � }|j dk rd|t j |� � < nt j |� � sd}|�/t j |� � }|t j | |z
� � z }nt j | |z
� � }t j
d� � � 5 t j |||�� � }|rt j |� � }||z }t j
|� � } ddd� � n # 1 swxY w Y |st j ||�
� � }| |z
} |r| |fS | S ) a�
Compute the log of the sum of exponentials of input elements.
Parameters
----------
a : array_like
Input array.
axis : None or int or tuple of ints, optional
Axis or axes over which the sum is taken. By default `axis` is None,
and all elements are summed.
.. versionadded:: 0.11.0
b : array-like, optional
Scaling factor for exp(`a`) must be of the same shape as `a` or
broadcastable to `a`. These values may be negative in order to
implement subtraction.
.. versionadded:: 0.12.0
keepdims : bool, optional
If this is set to True, the axes which are reduced are left in the
result as dimensions with size one. With this option, the result
will broadcast correctly against the original array.
.. versionadded:: 0.15.0
return_sign : bool, optional
If this is set to True, the result will be a pair containing sign
information; if False, results that are negative will be returned
as NaN. Default is False (no sign information).
.. versionadded:: 0.16.0
Returns
-------
res : ndarray
The result, ``np.log(np.sum(np.exp(a)))`` calculated in a numerically
more stable way. If `b` is given then ``np.log(np.sum(b*np.exp(a)))``
is returned.
sgn : ndarray
If return_sign is True, this will be an array of floating-point
numbers matching res and +1, 0, or -1 depending on the sign
of the result. If False, only one result is returned.
See Also
--------
numpy.logaddexp, numpy.logaddexp2
Notes
-----
NumPy has a logaddexp function which is very similar to `logsumexp`, but
only handles two arguments. `logaddexp.reduce` is similar to this
function, but may be less stable.
Examples
--------
>>> import numpy as np
>>> from scipy.special import logsumexp
>>> a = np.arange(10)
>>> logsumexp(a)
9.4586297444267107
>>> np.log(np.sum(np.exp(a)))
9.4586297444267107
With weights
>>> a = np.arange(10)
>>> b = np.arange(10, 0, -1)
>>> logsumexp(a, b=b)
9.9170178533034665
>>> np.log(np.sum(b*np.exp(a)))
9.9170178533034647
Returning a sign flag
>>> logsumexp([1,2],b=[1,-1],return_sign=True)
(1.5413248546129181, -1.0)
Notice that `logsumexp` does not directly support masked arrays. To use it
on a masked array, convert the mask into zero weights:
>>> a = np.ma.array([np.log(2), 2, np.log(3)],
... mask=[False, True, False])
>>> b = (~a.mask).astype(int)
>>> logsumexp(a.data, b=b), np.log(5)
1.6094379124341005, 1.6094379124341005
F�� check_finiteNr g T��axis�keepdims�ignore��divide)r )r �np�broadcast_arrays�any�inf�amax�ndim�isfinite�asarray�exp�errstate�sum�sign�log�squeeze)
�ar �br � return_sign�a_max�tmp�s�sgn�outs
�:/usr/lib/python3/dist-packages/scipy/special/_logsumexp.pyr r s� � �l �1�5�1�1�1�A��}��"�1�a�(�(���1�
�6�!�q�&�>�>� ��B��A����A�a�1�f�I� �G�A�D�4� 0� 0� 0�E� �z�A�~�~�%&��r�{�5�!�!�!�"�"�
�[��
�
� ����}� �J�q�M�M���"�&��U��#�#�#����f�Q��Y� � ��
��H� %� %� %� � � �F�3�T�H� 5� 5� 5�� � ��'�!�*�*�C�
��H�A��f�Q�i�i�� � � � � � � � � � � ���� � � � � � -��
�5�t�,�,�,���5�L�C�� ��C�x���
s � AE � E"�%E"c � � t | d�� � } t j | |d�� � }t j | |z
� � }|t j ||d�� � z S )a�
Compute the softmax function.
The softmax function transforms each element of a collection by
computing the exponential of each element divided by the sum of the
exponentials of all the elements. That is, if `x` is a one-dimensional
numpy array::
softmax(x) = np.exp(x)/sum(np.exp(x))
Parameters
----------
x : array_like
Input array.
axis : int or tuple of ints, optional
Axis to compute values along. Default is None and softmax will be
computed over the entire array `x`.
Returns
-------
s : ndarray
An array the same shape as `x`. The result will sum to 1 along the
specified axis.
Notes
-----
The formula for the softmax function :math:`\sigma(x)` for a vector
:math:`x = \{x_0, x_1, ..., x_{n-1}\}` is
.. math:: \sigma(x)_j = \frac{e^{x_j}}{\sum_k e^{x_k}}
The `softmax` function is the gradient of `logsumexp`.
The implementation uses shifting to avoid overflow. See [1]_ for more
details.
.. versionadded:: 1.2.0
References
----------
.. [1] P. Blanchard, D.J. Higham, N.J. Higham, "Accurately computing the
log-sum-exp and softmax functions", IMA Journal of Numerical Analysis,
Vol.41(4), :doi:`10.1093/imanum/draa038`.
Examples
--------
>>> import numpy as np
>>> from scipy.special import softmax
>>> np.set_printoptions(precision=5)
>>> x = np.array([[1, 0.5, 0.2, 3],
... [1, -1, 7, 3],
... [2, 12, 13, 3]])
...
Compute the softmax transformation over the entire array.
>>> m = softmax(x)
>>> m
array([[ 4.48309e-06, 2.71913e-06, 2.01438e-06, 3.31258e-05],
[ 4.48309e-06, 6.06720e-07, 1.80861e-03, 3.31258e-05],
[ 1.21863e-05, 2.68421e-01, 7.29644e-01, 3.31258e-05]])
>>> m.sum()
1.0
Compute the softmax transformation along the first axis (i.e., the
columns).
>>> m = softmax(x, axis=0)
>>> m
array([[ 2.11942e-01, 1.01300e-05, 2.75394e-06, 3.33333e-01],
[ 2.11942e-01, 2.26030e-06, 2.47262e-03, 3.33333e-01],
[ 5.76117e-01, 9.99988e-01, 9.97525e-01, 3.33333e-01]])
>>> m.sum(axis=0)
array([ 1., 1., 1., 1.])
Compute the softmax transformation along the second axis (i.e., the rows).
>>> m = softmax(x, axis=1)
>>> m
array([[ 1.05877e-01, 6.42177e-02, 4.75736e-02, 7.82332e-01],
[ 2.42746e-03, 3.28521e-04, 9.79307e-01, 1.79366e-02],
[ 1.22094e-05, 2.68929e-01, 7.31025e-01, 3.31885e-05]])
>>> m.sum(axis=1)
array([ 1., 1., 1.])
Fr Tr
)r r r r r )�xr �x_max�
exp_x_shifteds r& r r � s\ � �v �1�5�1�1�1�A� �G�A�D�4� 0� 0� 0�E��F�1�u�9�%�%�M� �2�6�-�d�T�J�J�J� J�J� c � � t | d�� � } t j | |d�� � }|j dk rd|t j |� � < nt j |� � sd}| |z
}t j |� � }t j d�� � 5 t j ||d�� � }t j |� � }ddd� � n # 1 swxY w Y ||z
}|S ) a� Compute the logarithm of the softmax function.
In principle::
log_softmax(x) = log(softmax(x))
but using a more accurate implementation.
Parameters
----------
x : array_like
Input array.
axis : int or tuple of ints, optional
Axis to compute values along. Default is None and softmax will be
computed over the entire array `x`.
Returns
-------
s : ndarray or scalar
An array with the same shape as `x`. Exponential of the result will
sum to 1 along the specified axis. If `x` is a scalar, a scalar is
returned.
Notes
-----
`log_softmax` is more accurate than ``np.log(softmax(x))`` with inputs that
make `softmax` saturate (see examples below).
.. versionadded:: 1.5.0
Examples
--------
>>> import numpy as np
>>> from scipy.special import log_softmax
>>> from scipy.special import softmax
>>> np.set_printoptions(precision=5)
>>> x = np.array([1000.0, 1.0])
>>> y = log_softmax(x)
>>> y
array([ 0., -999.])
>>> with np.errstate(divide='ignore'):
... y = np.log(softmax(x))
...
>>> y
array([ 0., -inf])
Fr Tr
r r
r N) r r r r r r r r r )r( r r) r" �exp_tmpr# r% s r& r r � s � �h �1�5�1�1�1�A� �G�A�D�4� 0� 0� 0�E� �z�A�~�~�%&��r�{�5�!�!�!�"�"�
�[��
�
� ���
�e�)�C��f�S�k�k�G�
��H� %� %� %� � � �F�7��� 5� 5� 5���f�Q�i�i��� � � � � � � � � � ���� � � � � ��)�C� �Js �,C � C
�C
)NNFF)N)�numpyr �scipy._lib._utilr �__all__r r r � r+ r& �<module>r2 s� �� � � � � /� /� /� /� /� /�
1�
1�
1��y� y� y� y�x^K� ^K� ^K� ^K�BF� F� F� F� F� Fr+