Implement __rpow__ for Tensor (#2731)

### Changes Implement __rpow__ for Tensor ### Related tickets 143943 ### Tests TemplateTestNNCFTensorOperators :: test_operators_int_rev
openvinotoolkit · Jun 13, 2024 · b780041 · b780041
1 parent d06b174
commit b780041
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/nncf/experimental/tensor/tensor.py b/nncf/experimental/tensor/tensor.py
@@ -97,6 +97,9 @@ def __rmul__(self, other: Union[Tensor, float]) -> Tensor:
     def __pow__(self, other: Union[Tensor, float]) -> Tensor:
         return Tensor(self.data ** unwrap_tensor_data(other))
 
+    def __rpow__(self, other: Union[Tensor, float]) -> Tensor:
+        return Tensor(unwrap_tensor_data(other) ** self.data)
+
     def __truediv__(self, other: Union[Tensor, float]) -> Tensor:
         return _call_function("_binary_op_nowarn", self, other, operator.truediv)
 

diff --git a/tests/shared/test_templates/template_test_nncf_tensor.py b/tests/shared/test_templates/template_test_nncf_tensor.py
@@ -37,6 +37,7 @@
     "floordiv": operator.floordiv,
     "neg": lambda a, _: -a,
 }
+BINARY_OPERATORS = ["add", "sub", "pow", "mul", "truediv", "floordiv"]
 
 COMPARISON_OPERATOR_MAP = {
     "lt": operator.lt,
@@ -121,7 +122,7 @@ def test_operators_int(self, op_name):
         assert isinstance(res_nncf, Tensor)
         assert res_nncf.device == nncf_tensor_a.device
 
-    @pytest.mark.parametrize("op_name", ("add", "sub", "mul", "truediv", "floordiv"))
+    @pytest.mark.parametrize("op_name", BINARY_OPERATORS)
     def test_operators_int_rev(self, op_name):
         tensor_a = self.to_tensor([1, 2])
         value = 2