slim.softmax()

2022-09-04 21:25:16 浏览数 (1)

代码语言:javascript复制
softmax(logits, scope=None)

对n维logit张量的第n维执行softmax。对于二维logits,这可以归结为tf.n .softmax。第n个维度需要具有指定数量的元素(类的数量)。

参数:

  • logits: N维张量,其中N > 1。
  • scope:variable_scope的可选作用域。

返回值:

  • 一个形状和类型与logits相同的“张量”。

0 人点赞