gpt4 book ai didi

python - 具有元组输入的自定义 TensorFlow RNN 单元

转载 作者:太空宇宙 更新时间:2023-11-04 04:53:56 24 4
gpt4 key购买 nike

我试图在 TensorFlow 中创建一个接受元组作为输入的自定义 RNN 单元,但我遇到了父类 BasicLSTMCell 要求输入是二维的问题:

# Inputs must be 2-dimensional.
self.input_spec = base_layer.InputSpec(ndim=2)

我怎样才能绕过这个限制?我无法在 call() 方法中添加处理元组的逻辑,因为执行永远不会到达该方法 - 维度检查会引发错误。

最佳答案

其实我也发现了这个问题。 tensorflow 平台存在一个错误。可以通过更改recurrent.py文件中的get_step_input_shape函数来解决。只需将 [0] 添加到此行的末尾:nest.map_structure(get_input_spec, input_shape))

关于python - 具有元组输入的自定义 TensorFlow RNN 单元,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47520135/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com