gpe.gpu.bec2
============

.. py:module:: gpe.gpu.bec2

.. autoapi-nested-parse::

   GPU versions of the code.

   Requires cupy and the various NVIDIA dependencies.



Attributes
----------

.. autoapisummary::

   gpe.gpu.bec2.u


Classes
-------

.. autoapisummary::

   gpe.gpu.bec2.State


Module Contents
---------------

.. py:data:: u

.. py:class:: State(basis=None, Nxyz=(2**5, 2**5, 2**5), Lxyz=(30 * u.micron, 50 * u.micron, 50 * u.micron), symmetric_grid=False, twist=None, **kw)

   Bases: :py:obj:`pytimeode.mixins_gpu.GPUArrayStateMixin`, :py:obj:`gpe.bec2.State`


   Mixin providing support for states with a single GPU data array.


   .. py:attribute:: xp


   .. py:attribute:: asnumpy