blob: b51b468c3da79be21da4423c9363f262ba7bfd2c (
plain)
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
}
|
