python - 将 CudaNdarraySharedVariable 转换为 TensorVariable

Question

我正在尝试将pylearn2GPU 模型转换为 CPU 兼容版本以在远程服务器上进行预测——如何将CudaNdarraySharedVariable's 转换为TensorVariable's 以避免在无 GPU 机器上调用 cuda 代码时出错？实验性的 theano 标志unpickle_gpu_to_cpu似乎留下了一些CudaNdarraySharedVariable（特别是model.layers[n].transformer._W）。

Answer 1

对于一个普通的 CudaNdarray 变量，这样的东西应该可以工作：

'''x = CudaNdarray... x_new=theano.tensor.TensorVariable(CudaNdarrayType([False] * tensor_dim))
f = theano.function([x_new], x_new)

转换_x = f(x) '''