fix arg_shapes of LaplaceTerm