def testSumStates(self):
idx = [
[0, 1, 0, 1],
[0, 0, 0, 1],
]
states = math_ops.log([
[[1.0, 2.0, 3.0, 4.0], [5.0, 6.0, 7.0, 8.0]],
[[0.1, 0.2, 0.3, 0.4], [0.5, 0.6, 0.7, 0.8]],
])
sum_of_states = math_ops.exp(ctc_ops._sum_states(idx, states))
self.assertAllClose([[[4.0, 6.0, 0.0, 0.0], [18.0, 8.0, 0.0, 0.0]],
[[0.4, 0.6, 0.0, 0.0], [1.8, 0.8, 0.0, 0.0]]],
sum_of_states)
def testSumStates(self):
idx = [
[0, 1, 0, 1],
[0, 0, 0, 1],
]
states = math_ops.log([
[[1.0, 2.0, 3.0, 4.0],
[5.0, 6.0, 7.0, 8.0]],
[[0.1, 0.2, 0.3, 0.4],
[0.5, 0.6, 0.7, 0.8]],
])
sum_of_states = math_ops.exp(ctc_ops._sum_states(idx, states))
self.assertAllClose([
[[4.0, 6.0, 0.0, 0.0],
[18.0, 8.0, 0.0, 0.0]],
[[0.4, 0.6, 0.0, 0.0],
[1.8, 0.8, 0.0, 0.0]]
], sum_of_states)
