def bptrs(a):
 return pycuda.gpuarray.arange(a.ptr, (a.ptr + (a.shape[0] * a.strides[0])), a.strides[0], dtype=cublas.ctypes.c_void_p)

def bptrs ( a ) : return pycuda . gpuarray . arange ( a . ptr , ( a . ptr + ( a . shape [ 0 ] * a . strides [ 0 ] ) ) , a . strides [ 0 ] , dtype = cublas . ctypes . c_void_p )
pointer array when input represents a batch of matrices .