Show tensor dimensions in variable explorer (microsoft#14244)

Author: joyceerhl

pythonFiles/tests/ipython/getJupyterVariableList.py

@@ -5,7 +5,9 @@
 from IPython import get_ipython as _VSCODE_get_ipython
 
 # _VSCode_supportsDataExplorer will contain our list of data explorer supported types
-_VSCode_supportsDataExplorer = "['list', 'Series', 'dict', 'ndarray', 'DataFrame']"
+_VSCode_supportsDataExplorer = (
+    "['list', 'Series', 'dict', 'ndarray', 'DataFrame', 'Tensor']"
+)
 
 # who_ls is a Jupyter line magic to fetch currently defined vars
 _VSCode_JupyterVars = _VSCODE_get_ipython().run_line_magic("who_ls", "")

pythonFiles/vscode_datascience_helpers/dataframes/vscodeGetVariableInfo.py

@@ -5,6 +5,21 @@
 import builtins as _VSCODE_builtins
 import vscodeDataFrameHelpers as _VSCODE_dataFrameHelpers
 
+
+def _VSCODE_maybeParseTensorShape(var, result):
+    try:
+        vartype = type(var)
+        if (hasattr(vartype, "__name__")) and vartype.__name__ == "Tensor":
+            varshape = str(var.shape)
+            start = varshape.index("[")
+            end = varshape.index("]")
+            if start > 0 and end > 0:
+                res = "(" + varshape[start + 1 : end] + ")"
+                result["shape"] = res
+    except TypeError:
+        pass
+
+
 # Function to do our work. It will return the object
 def _VSCODE_getVariableInfo(var):
     # Start out without the information
@@ -26,6 +41,7 @@ def _VSCODE_getVariableInfo(var):
                 ):
                     result["shape"] = _VSCODE_shapeStr
                 del _VSCODE_shapeStr
+            _VSCODE_maybeParseTensorShape(var, result)
         except TypeError:
             pass
 

pythonFiles/vscode_datascience_helpers/getJupyterVariableDataFrameInfo.py

@@ -5,7 +5,9 @@
 import builtins as _VSCODE_builtins
 
 # _VSCode_sub_supportsDataExplorer will contain our list of data explorer supported types
-_VSCode_supportsDataExplorer = "['list', 'Series', 'dict', 'ndarray', 'DataFrame']"
+_VSCode_supportsDataExplorer = (
+    "['list', 'Series', 'dict', 'ndarray', 'DataFrame', 'Tensor']"
+)
 
 # In IJupyterVariables.getValue this '_VSCode_JupyterTestValue' will be replaced with the json stringified value of the target variable
 # Indexes off of _VSCODE_targetVariable need to index types that are part of IJupyterVariable

src/client/datascience/jupyter/kernelVariables.ts

@@ -34,7 +34,7 @@ const DocStringRegex = /.*?\[.*?;31mDocstring:.*?\[0m\s+(.*)/;
 const CountRegex = /.*?\[.*?;31mLength:.*?\[0m\s+(.*)/;
 const ShapeRegex = /^\s+\[(\d+) rows x (\d+) columns\]/m;
 
-const DataViewableTypes: Set<string> = new Set<string>(['DataFrame', 'list', 'dict', 'ndarray', 'Series']);
+const DataViewableTypes: Set<string> = new Set<string>(['DataFrame', 'list', 'dict', 'ndarray', 'Series', 'Tensor']);
 
 interface INotebookState {
     currentExecutionCount: number;