Exemplo n.º 1
0
def test_print_embedding():
    net = Embedding(20000, 768, False)
    print(net)
Exemplo n.º 2
0
def test_check_embedding_3():
    net = Embedding(20000, 768, True, "zeros")
    input_data = Tensor(np.ones([8, 128]), dtype.int32)
    _executor.compile(net, input_data)