Add softmax into Python API.

emailweixu-patch-1
dangqingqing 7 years ago
parent 292c1951f0
commit 4673a24bda

@ -59,6 +59,7 @@ __all__ = [
'elementwise_pow',
'clip',
'clip_by_norm',
'softmax',
'sequence_softmax',
] + __activations__

@ -223,6 +223,14 @@ class TestBook(unittest.TestCase):
self.assertIsNotNone(layers.sequence_softmax(x=seq))
print(str(program))
def test_softmax(self):
program = Program()
with program_guard(program):
data = layers.data(name='data', shape=[10], dtype='float32')
hid = layers.fc(input=data, size=20)
self.assertIsNotNone(layers.softmax(x=hid))
print(str(program))
def test_get_places(self):
program = Program()
with program_guard(program):

Loading…
Cancel
Save