Ejemplo n.º 1
0
    def test_custom_name(self):
        layer = tl.Layer()
        self.assertIn('Layer', str(layer))
        self.assertNotIn('CustomLayer', str(layer))

        layer = tl.Layer(name='CustomLayer')
        self.assertIn('CustomLayer', str(layer))
Ejemplo n.º 2
0
 def test_init_returns_empty_weights_and_state(self):
     layer = tl.Layer()
     input_signature = shapes.ShapeDtype((2, 5))
     weights, state = layer.init(input_signature)
     self.assertEmpty(weights)
     self.assertEmpty(state)
Ejemplo n.º 3
0
 def test_forward_raises_error(self):
     layer = tl.Layer()
     x = np.array([[1, 2, 3, 4, 5], [10, 20, 30, 40, 50]])
     with self.assertRaises(NotImplementedError):
         _ = layer.forward(x)
Ejemplo n.º 4
0
 def test_set_weighs_raises_error(self):
     layer = tl.Layer()
     layer.weights = 1.0  # can assign weights
     with self.assertRaisesRegex(ValueError, 'weighs'):
         layer.weighs = 1.0  # cannot assign weighs
Ejemplo n.º 5
0
 def test_call_raises_error(self):
     layer = tl.Layer()
     x = np.array([[1, 2, 3, 4, 5], [10, 20, 30, 40, 50]])
     with self.assertRaisesRegex(tl.LayerError, 'NotImplementedError'):
         _ = layer(x)