tensorflow2.0——softmax函数

import tensorflow as tf

logits = tf.random.uniform([1,10],minval=-2,maxval=2)
print('logits:',logits)

prob = tf.nn.softmax(logits)
print('softmax后:',prob)

p_sum = tf.reduce_sum(prob)
print('相加值:',p_sum)

 

posted @ 2020-08-15 17:49  山…隹  阅读(834)  评论(0编辑  收藏  举报