tf.keras.layers.GRUCell

`tf.keras.layers.GRUCell`是TensorFlow中用于构建GRU（Gated Recurrent Unit）单元的类。GRU是一种循环神经网络（RNN）的变体，用于处理序列数据。

GRU单元是一种特殊的RNN单元，它具有更新门和重置门。这些门控制着信息在序列中的流动，使得GRU能够更好地捕捉长期依赖关系。GRU单元的输出可以作为下一个时间步的输入，从而实现对序列数据的建模。

以下是使用`tf.keras.layers.GRUCell`的示例代码：

import tensorflow as tf

# 创建GRUCell对象
gru_cell = tf.keras.layers.GRUCell(units)

# 在时间轴上循环展开运算
inputs = tf.zeros([2, 50, 64])
rnn = tf.keras.layers.RNN(gru_cell)
output = rnn(inputs)

print(output.shape)  # 输出：(2, units)

在上面的示例中，我们首先创建了一个`GRUCell`对象，并将其传递给`RNN`层。然后，我们将输入数据`inputs`传递给`RNN`层，得到输出`output`。输出的形状是`(2, units)`，其中`units`是GRU单元的输出维度。