code: def test_flatten():
      m = theano.tensor.fmatrix()
      f = theano.function([m], m.flatten(), mode=mode_with_gpu)
      val = np.random.rand(10, 11).astype("float32")
      res = f(val)
      utt.assert_allclose(res, val.flatten())
      assert res.shape == val.flatten().shape
      assert GpuReshape in [
          type(node.op) for node in f.maker.fgraph.toposort()
      ]
      val = np.random.rand(10, 11).astype("float32")
      res = f(val)
      utt.assert_allclose(res, val.flatten())
      assert res.shape == val.flatten().shape
      assert GpuReshape in [
          type(node.op) for node in f.maker.fgraph.toposort()
      ]
      assert _check_stack_trace(f)
  
      f = theano.function(
          [m],
          m.flatten(ndim=2),
          mode=mode_with_gpu.excluding("local_useless_reshape"),
      )
      val = np.random.rand(10, 11).astype("float32")
      res = f(val)
      utt.assert_allclose(res, val)
      assert res.shape == val.shape
      assert GpuReshape in [
          type(node.op) for node in f.maker.fgraph.toposort()
      ]
      assert _check_stack_trace(f)
  
      m = theano.tensor.tensor3()
      f = theano.function([m], m.flatten(ndim=2), mode=mode_with_gpu)
      val = np.random.rand(10, 11, 12).astype("float32")
      res = f(val)
      utt.assert_allclose(res, val.reshape(10, -1))
      assert res.shape == val.reshape(10, -1).shape
      assert GpuReshape in [
          type(node.op) for node in f.maker.fgraph.toposort()
      ]
      assert _check_stack_trace(f)
  
imports: from __future__ import absolute_import, print_function, division
  from nose.tools import assert_raises
  import numpy
  import theano
  from theano import tensor
  import theano.tensor.slinalg
  from theano.tests.breakpoint import PdbBreakpoint
  from theano.tests import unittest_tools, test_ifelse
  from theano.tensor.tests import test_basic
  from theano.gof.opt import check_stack_trace
  import theano.gpuarray
  from ..None import basic_ops
  from ..type import GpuArrayType, gpuarray_shared_constructor, get_context
  from ..basic_ops import GpuAlloc, GpuAllocEmpty, GpuReshape, GpuFromHost, HostFromGpu, host_from_gpu
  from ..blas import GpuGemm
  from ..elemwise import GpuCAReduceCuda, GpuCAReduceCPY, GpuElemwise, Elemwise, max_inputs_to_GpuElemwise
  from ..dnn import GpuDnnReduction
  from ..subtensor import GpuSubtensor
  from ..linalg import GpuCusolverSolve, cusolver_available, GpuCholesky
  from .config import mode_with_gpu, mode_without_gpu, test_ctx_name, SkipTest
  import unittest
  from theano.tensor.nnet import abstract_conv
  from theano.gpuarray import dnn, blas, opt
label_desc: Use image node_id 6 for calling a global function with example usage: test_flatten() without return types