修复当前Google Colab TF版本中没有模块的“图层”

Question

我使用Google colab（python3 GPU）

例如，我想运行this repo codes，但在这些行中运行demo ipynb时出现错误：

import tensorflow as tf
from layers import (_causal_linear, _output_linear, conv1d, dilated_conv1d) 

运行这两行时，出现错误“无模块层”

我认为这不是bug或其他内容，因为此回购协议已超过1000颗星。 我认为这是一个tf版本问题。

有什么办法解决这个问题吗？

Answer 1

“层”是您在Tensorflow中未链接的包中的模块。参见here。

顺便说一句，如果您想从张量流子模块中导入，则必须执行from tensorflow.package import ...。