与gpuArray合作的theano'as_op'

Question

我正在研究一个简单的theano as_op：

@as_op(itypes=[theano.tensor.lmatrix, theano.tensor.lvector, theano.tensor.lvector], otypes=[theano.tensor.lmatrix])#, infer_shape=infer_shape_numpy)
def increment_A(A, bmus_i0, i1):
    np.add.at(A, (bmus_i0, i1), 1)
    np.add.at(A, (i1, bmus_i0), 1)
    return A

只有当我使用cpu时才有效，当我尝试使用gpu时，我会收到错误

TypeError: We expected inputs of types '[TensorType(int64, matrix), TensorType(int64, vector), TensorType(int64, vector)]' but got types '[GpuArrayType<None>(int64, matrix), TensorType(int64, vector), TensorType(int64, vector)]'

当然它需要gpuarray作为输入...... 我怎样才能正确设置方法的输入？