gistfile1.py

import theano
import theano.tensor as T
import theano.sandbox.cuda as cuda
from theano.printing import pydotprint
from theano.misc import pycuda_utils
from pycuda.gpuarray import zeros
import pycuda.driver as drv
import pycuda.autoinit
from pycuda.compiler import SourceModule


X = cuda.fmatrix()
A = cuda.vector()

indexed = X[T.cast(A, 'int32')]
f = theano.function([X, A], indexed)

x = pycuda_utils.to_cudandarray(zeros((2, 2), dtype='float32'))
a = pycuda_utils.to_cudandarray(zeros((4,), dtype='float32'))

pydotprint(f, 'simple.png')

print f(x, a)