标签: keras plaidml
我目前正在查看PlaidML Keras后端。
我了解使用PlaidML张量操作是在切片代码中内部编程的。
有人可以告诉我,keras模型是如何在内部映射到切片语言的吗?
说,我的神经网络有100层,理论上可以完全用图块语言编写。 (矩阵乘法和S型函数)Keras后端是否将整个网络“编译”为单个整体图块函数,或者每个层和操作都被编译为自己的图块函数?
我怀疑看看代码,情况就是这样。如果是这样,那么在CPU上的python和GPU上的操作之间进行上下文切换不是瓶颈吗?