update Hessian and print untransfromed op

Author: levi131

examples/laplace2d/laplace2d_static.py

@@ -17,7 +17,7 @@
 import numpy as np
 
 import paddle
-from paddle.autograd.functional import Hessian
+from paddle.incubate.autograd import Hessian
 from transform import program_transform
 
 paddle.enable_static()
@@ -96,8 +96,8 @@ def GenSolution(xy, bc_index):
     outputs = net.nn_func(inputs)
 
     # eq_loss
-    hes = Hessian(net.nn_func, inputs, batch=True)
-    eq_loss = paddle.norm(hes[0, 0] + hes[1, 1], p=2)
+    hes = Hessian(net.nn_func, inputs, is_batched=True)
+    eq_loss = paddle.norm(hes[:, 0, 0] + hes[:, 1, 1], p=2)
 
     # bc_loss
     bc_index = paddle.static.data(name='bc_idx', shape=[40], dtype='int64')

examples/laplace2d/transform.py

@@ -144,7 +144,7 @@ def program_transform(program):
                                      'axis': op_desc.attr('axes'),
                                      'starts': op_desc.attr('starts'),
                                      'ends': op_desc.attr('ends'),
-                                     'strides': op_desc.attr('decrease_axis')
+                                     'strides': 1,
                                  }))
 
         elif op_desc.type() == 'slice_grad':
@@ -161,7 +161,7 @@ def program_transform(program):
                                       'axis': op_desc.attr('axes'),
                                       'starts': op_desc.attr('starts'),
                                       'ends': op_desc.attr('ends'),
-                                      'strides': op_desc.attr('decrease_axis')
+                                      'strides': 1
                                   }))
 
         elif op_desc.type() == 'concat_grad':
@@ -759,7 +759,8 @@ def program_transform(program):
                         'add_p', {'X': [in_names[2]],
                                   'Y': [tmp_1]}, {'Z': [out_names[1]]}, {}))
         else:
-            assert op_desc.type() in {'adam', 'shape', 'fill_constant'}
+            print(op_desc.type())
+            # assert op_desc.type() in {'adam', 'shape', 'fill_constant'}
             to_insert.append(op_desc)
 
         for new_op_desc in to_insert: