diff --git a/csrc/python_frontend/python_bindings.cpp b/csrc/python_frontend/python_bindings.cpp index ab0123d496e..d20105e990a 100644 --- a/csrc/python_frontend/python_bindings.cpp +++ b/csrc/python_frontend/python_bindings.cpp @@ -997,7 +997,13 @@ void initNvFuserPythonBindings(PyObject* module) { return ss.str(); }); tensor_class.def_property_readonly( - "ndim", [](Tensor& self) { return self.dims; }); + "ndim", + [](Tensor& self) { return self.dims; }, + "Returns the rank of the tensor."); + tensor_class.def_property_readonly( + "index", + [](Tensor& self) { return self.index; }, + "Returns the index of the tensor as in FusionDefinition.sched.tensors()."); tensor_class.def("_get_fusion_definition", [](Tensor& self) { return self.fusion_definition; });