layers_test.py 文件源码

def test_embedding_raises(self):
    self.assertRaises(ValueError, tdl.Embedding, 2, 2, np.zeros([3, 3]))
    six.assertRaisesRegex(
        self, TypeError, 'Embeddings take scalar inputs.', tdl.Embedding(2, 2),
        tf.constant([[0, 0]], 'int32'))
    six.assertRaisesRegex(
        self, TypeError, 'Embeddings take integer inputs.', tdl.Embedding(2, 2),
        tf.constant([0], 'float32'))