Пример #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)
Пример #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)))
Пример #3
0
def test_base_backend_random_uniforl_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.random_uniform((2, 2))
Пример #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())
Пример #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)))
Пример #7
0
def test_base_backend_expm_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.expm(np.ones((2, 2)))
Пример #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)))
Пример #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)
Пример #10
0
def test_base_backend_rq_decompositon_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.rq_decomposition(np.ones((2, 2)), 0)
Пример #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))
Пример #12
0
def test_base_backend_transpose_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.transpose(np.ones((2, 2)), [0, 1])
Пример #13
0
def test_base_backend_reshape_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.reshape(np.ones((2, 2)), (4, 1))
Пример #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]])
Пример #15
0
def test_base_backend_name():
    backend = BaseBackend()
    assert backend.name == "base backend"
Пример #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)))
Пример #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)))
Пример #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)))
Пример #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)))
Пример #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)))
Пример #21
0
def test_base_backend_sparse_shape_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.sparse_shape(np.ones((2, 2)))
Пример #22
0
def test_base_backend_einsul_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.einsum("ii", np.ones((2, 2)))
Пример #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)
Пример #24
0
def test_base_backend_eye_not_implemented():
    backend = BaseBackend()
    with pytest.raises(NotImplementedError):
        backend.eye(2, dtype=np.float64)