tf.compat.v1.math.softmax
Stay organized with collections
Save and categorize content based on your preferences.
Computes softmax activations. (deprecated arguments)
tf . compat . v1 . math . softmax (
logits , axis = None , name = None , dim = None
)
Warning: SOME ARGUMENTS ARE DEPRECATED: (dim)
. They will be removed in a future version.
Instructions for updating:
dim is deprecated, use axis instead
This function performs the equivalent of
softmax = tf . exp ( logits ) / tf . reduce_sum ( tf . exp ( logits ), axis )
See: https://en.wikipedia.org/wiki/Softmax_function
Example usage:
tf . nn . softmax ([ - 1 , 0. , 1. ])
<tf . Tensor : shape = ( 3 ,), dtype = float32 ,
numpy = array ([ 0.09003057 , 0.24472848 , 0.66524094 ], dtype = float32 ) >
Args
logits
A non-empty Tensor
, or an object whose type has a registered
Tensor
conversion function. Must be one of the following types:
half
,float32
, float64
. See also convert_to_tensor
axis
The dimension softmax would be performed on. The default is -1 which
indicates the last dimension.
name
A name for the operation (optional).
dim
Deprecated alias for axis
.
Returns
A Tensor
. Has the same type and shape as logits
.
Raises
InvalidArgumentError
if logits
is empty or axis
is beyond the last
dimension of logits
.
TypeError
If no conversion function is registered for logits
to
Tensor.
RuntimeError
If a registered conversion function returns an invalid
value.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License , and code samples are licensed under the Apache 2.0 License . For details, see the Google Developers Site Policies . Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license .
Last updated 2021-02-18 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Missing the information I need"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Too complicated / too many steps"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Out of date"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Samples / code issue"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Other"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Easy to understand"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Solved my problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Other"
}]
{"lastModified": "Last updated 2021-02-18 UTC."}
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2021-02-18 UTC."],[],[]]