标签： python tensorflow keras word-embedding
我很好奇Keras如何处理内部的CPU和GPU分配。不幸的是，在这方面，Keras网站上的文档非常有限。
我在内存中安装Embedding层时遇到了麻烦，因此我遵循了answer中的说明，仅将Embedding层强制安装在CPU上，其余部分（ GPU上的密集层等）。我现在没有任何问题，但很好奇它的工作原理。
Embedding
CPU和GPU上都存在图形版本吗？还是在完成嵌入后将数据传递到GPU上图形的其余部分？如果是这样，如何训练举重？