tensorflow/core/api_def/base_api/api_def_Softmax.pbtxt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

op {
  graph_op_name: "Softmax"
  in_arg {
    name: "logits"
    description: <<END
2-D with shape `[batch_size, num_classes]`.
END
  }
  out_arg {
    name: "softmax"
    description: <<END
Same shape as `logits`.
END
  }
  summary: "Computes softmax activations."
  description: <<END
For each batch `i` and class `j` we have

    softmax[i, j] = exp(logits[i, j]) / sum_j(exp(logits[i, j]))
END
}