Skip to content
This repository was archived by the owner on Nov 7, 2024. It is now read-only.

Commit d87ee43

Browse files
authored
temporarily switch to tf.tensordot instead of tensordot2 (#941)
* remove tensordot2; it has a bug * remove another tensordot2 call
1 parent edf3f34 commit d87ee43

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tensornetwork/backends/tensorflow/tensorflow_backend.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
from typing import Union
1717
from tensornetwork.backends import abstract_backend
1818
from tensornetwork.backends.tensorflow import decompositions
19-
from tensornetwork.backends.tensorflow import tensordot2
2019
import functools as fct
2120
import operator as op
2221

@@ -46,7 +45,7 @@ def __init__(self) -> None:
4645

4746
def tensordot(self, a: Tensor, b: Tensor,
4847
axes: Union[int, Sequence[Sequence[int]]]) -> Tensor:
49-
return tensordot2.tensordot(tf, a, b, axes)
48+
return tf.tensordot(a, b, axes)
5049

5150
def reshape(self, tensor: Tensor, shape: Tensor) -> Tensor:
5251
return tf.reshape(tensor, shape)
@@ -108,7 +107,7 @@ def convert_to_tensor(self, tensor: Tensor) -> Tensor:
108107
return result
109108

110109
def outer_product(self, tensor1: Tensor, tensor2: Tensor) -> Tensor:
111-
return tensordot2.tensordot(tf, tensor1, tensor2, 0)
110+
return tf.tensordot(tensor1, tensor2, 0)
112111

113112
#pylint: disable=unused-argument
114113
def einsum(self,

0 commit comments

Comments
 (0)