Ejemplo n.º 1
0
 def __init__(self,
              rank: int,
              dimension: int,
              name: Text,
              axis_names: List[Text],
              backend: base_backend.BaseBackend,
              dtype: Type[np.number] = np.float64) -> None:
     # TODO: Make this computation lazy, once Node doesn't require tensor
     # at instatiation.
     copy_tensor = self.make_copy_tensor(rank, dimension, dtype)
     copy_tensor = backend.convert_to_tensor(copy_tensor)
     super().__init__(copy_tensor, name, axis_names, backend)
Ejemplo n.º 2
0
def test_base_backend_eigs_lanczos_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.eigsh_lanczos(lambda x: x, [], np.ones((2)))
Ejemplo n.º 3
0
def test_base_backend_random_uniforl_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.random_uniform((2, 2))
Ejemplo n.º 4
0
def test_base_backend_zeros_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.zeros((2, 2), dtype=np.float64)
def test_flatten_edges_different_backend_raises_value_error(single_node_edge):
  node1 = single_node_edge.node
  node2 = tn.Node(np.random.rand(2, 2, 2))
  node2.backend = BaseBackend()
  with pytest.raises(ValueError):
    tn.flatten_edges(node1.get_all_edges() + node2.get_all_edges())
Ejemplo n.º 6
0
def test_base_backend_broadcast_left_multiplication_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.broadcast_left_multiplication(np.ones((2, 2)), np.ones((2, 2)))
Ejemplo n.º 7
0
def test_base_backend_expm_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.expm(np.ones((2, 2)))
Ejemplo n.º 8
0
def test_base_backend_divide_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.divide(np.ones((2, 2)), np.ones((2, 2)))
Ejemplo n.º 9
0
def test_base_backend_shape_concat_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.shape_concat([np.ones((2, 2)), np.ones((2, 2))], 0)
Ejemplo n.º 10
0
def test_base_backend_rq_decompositon_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.rq_decomposition(np.ones((2, 2)), 0)
Ejemplo n.º 11
0
def test_base_backend_slice_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.slice(np.ones((2, 2)), (0, 1), (1, 1))
Ejemplo n.º 12
0
def test_base_backend_transpose_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.transpose(np.ones((2, 2)), [0, 1])
Ejemplo n.º 13
0
def test_base_backend_reshape_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.reshape(np.ones((2, 2)), (4, 1))
Ejemplo n.º 14
0
def test_base_backend_tensordot_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.tensordot(np.ones((2, 2)), np.ones((2, 2)), axes=[[0], [0]])
Ejemplo n.º 15
0
def test_base_backend_name():
    backend = BaseBackend()
    assert backend.name == "base backend"
Ejemplo n.º 16
0
def test_base_backend_subtraction_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.subtraction(np.ones((2, 2)), np.ones((2, 2)))
Ejemplo n.º 17
0
def test_base_backend_multiply_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.multiply(np.ones((2, 2)), np.ones((2, 2)))
Ejemplo n.º 18
0
def test_base_backend_convert_to_tensor_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.convert_to_tensor(np.ones((2, 2)))
Ejemplo n.º 19
0
def test_base_backend_index_update_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.index_update(np.ones((2, 2)), np.ones((2, 2)), np.ones((2, 2)))
Ejemplo n.º 20
0
def test_base_backend_outer_product_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.outer_product(np.ones((2, 2)), np.ones((2, 2)))
Ejemplo n.º 21
0
def test_base_backend_sparse_shape_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.sparse_shape(np.ones((2, 2)))
Ejemplo n.º 22
0
def test_base_backend_einsul_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.einsum("ii", np.ones((2, 2)))
Ejemplo n.º 23
0
def test_switch_backend_raises_error(backend):
    a = tn.Node(np.random.rand(3, 3, 3))
    a.backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        tn.switch_backend({a}, backend)
Ejemplo n.º 24
0
def test_base_backend_eye_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.eye(2, dtype=np.float64)