py:优化printtensor的format

Author: miaobyte

excuter/op-mem-ompsimd/src/deepx/op/print.hpp

@@ -15,7 +15,11 @@ namespace deepx::op{
             string name=this->args[0];
             if (mem.existstensor(name)){
                 auto t=mem.gettensor<T>(name);
-                tensorfunc::print<T>(*t);
+                if (this->args.size() == 1){
+                    tensorfunc::print<T>(*t);
+                }else{
+                    tensorfunc::print<T>(*t, this->args[1]);
+                }
             }else{
                 cout<<"<print> "<<name<<" not found"<<endl;
             }

excuter/op-mem-ompsimd/src/deepx/tensorfunc/print.hpp

@@ -31,11 +31,11 @@ namespace deepx::tensorfunc
             }
             else if constexpr (std::is_same_v<T, float>)
             {
-                format = "%.2f";
+                format = "%.4f";
             }
             else if constexpr (std::is_same_v<T, double>)
             {
-                format = "%.2f";    
+                format = "%.4f";    
             }
         }
         t.shape.print();

front/py/deepx/nn/functional/print.py

@@ -4,8 +4,8 @@
 from deepx.scheduler import send
 
 OpNode.register("print")
-def printtensor(t:Tensor):
-    ir=DeepxIR("print",'', [t.node.name], [])
+def printtensor(t:Tensor,format=''):
+    ir=DeepxIR("print",'', [t.node.name,format], [])
     send(ir)
     return ''
 

front/py/examples/2_ir/6_math.dot

@@ -0,0 +1,23 @@
+// Computational Graph
+digraph {
+	rankdir=TB
+	node [shape=record]
+	132560372090576 [label="t
+(10, 10)" color=skyblue fillcolor=aliceblue fontname="Sans-Serif" labeljust=l shape=box style=filled]
+	132560372104592 [label="var_1
+-1" color=orange fillcolor=moccasin fontname="Sans-Serif" labeljust=l shape=box style=filled]
+	132558398381392 [label="var_2
+1" color=orange fillcolor=moccasin fontname="Sans-Serif" labeljust=l shape=box style=filled]
+	132558397716416 [label=uniform color=darkslategray fillcolor=lightgray fontname="Courier Bold" labeljust=l shape=box style=filled]
+	132558396599424 [label="relu_t
+(10, 10)" color=skyblue fillcolor=aliceblue fontname="Sans-Serif" labeljust=l shape=box style=filled]
+	132558396599568 [label=max_scalar color=darkslategray fillcolor=lightgray fontname="Courier Bold" labeljust=l shape=box style=filled]
+	132558396599808 [label="var_3
+0" color=orange fillcolor=moccasin fontname="Sans-Serif" labeljust=l shape=box style=filled]
+	132558397716416 -> 132560372090576 [arrowsize=0.8 color=gray40 penwidth=1.2]
+	132560372104592 -> 132558397716416 [arrowsize=0.8 color=gray40 penwidth=1.2]
+	132558398381392 -> 132558397716416 [arrowsize=0.8 color=gray40 penwidth=1.2]
+	132558396599568 -> 132558396599424 [arrowsize=0.8 color=gray40 penwidth=1.2]
+	132560372090576 -> 132558396599568 [arrowsize=0.8 color=gray40 penwidth=1.2]
+	132558396599808 -> 132558396599568 [arrowsize=0.8 color=gray40 penwidth=1.2]
+}