Change .shape type to SimpleShape in Python

tenstorrent · Jan 23, 2025 · f3637a3 · f3637a3
1 parent 7c8282e
commit f3637a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ttnn/cpp/pybind11/pytensor.cpp b/ttnn/cpp/pybind11/pytensor.cpp
@@ -989,7 +989,7 @@ void pytensor_module(py::module& m_tensor) {
                     ttnn.Tensor(py_tensor, ttnn.bfloat16, device, ttnn.TILE_LAYOUT)
             )doc")
         .def_property_readonly("spec", [](const Tensor& self) { return self.get_tensor_spec(); })
-        .def_property_readonly("shape", [](const Tensor& self) { return self.get_shape(); })
+        .def_property_readonly("shape", [](const Tensor& self) { return self.get_logical_shape(); })
         .def_property_readonly("logical_shape", [](const Tensor& self) { return self.get_logical_shape(); })
         .def_property_readonly("dtype", [](const Tensor& self) { return self.get_dtype(); })
         .def_property_readonly("layout", [](const Tensor& self) { return self.get_layout(); })