def test_keras_import(self):
model = Sequential()
model.add(Permute((2, 1), input_shape=(10, 64)))
model.build()
self.keras_type_test(model, 0, 'Permute')
微信
公众号
扫码关注公众号