code: def square(x):
      """Elementwise square function.
  
      .. math::
         y_i = x_i ^ 2.
  
      Args:
          x (:class:`~chainer.Variable` or :ref:`ndarray`): Input variable.
              A :math:`(s_1, s_2, ..., s_N)` -shaped float array.
  
      Returns:
          ~chainer.Variable: Output variable.
          A :math:`(s_1, s_2, ..., s_N)` -shaped float array.
  
      .. admonition:: Example
  
          >>> x = np.arange(6).reshape(2,3).astype(np.float32)
          >>> x
          array([[0., 1., 2.],
                 [3., 4., 5.]], dtype=float32)
          >>> y = F.square(x)
          >>> y.shape
          (2, 3)
          >>> y.array
          array([[ 0.,  1.,  4.],
                 [ 9., 16., 25.]], dtype=float32)
  
      """
      return Square().apply((x,))[0]
  
file_path: chainer/chainer/functions/math/square.py
imports: from chainer import backend
  from chainer import function_node
  from chainer import utils
  from chainer.utils import type_check
label_desc: Use image node_id 1 for calling a global function with example usage: square(x) and returns: Square