sandbox.cuda.var – The Variables for Cuda-allocated arrays

API

class theano.sandbox.cuda.var.CudaNdarraySharedVariable(name, type, value, strict, allow_downcast=None, container=None)[source]

Shared Variable interface to CUDA-allocated arrays.

get_value(borrow=False, return_internal_type=False)[source]

Return the value of this SharedVariable’s internal array.

set_value(value, borrow=False)[source]

Assign value to the GPU-allocated array.