tf.raw_ops.LSTMBlockCell

Computes the LSTM cell forward propagation for 1 time step.

View aliases

Compat aliases for migration

tf.compat.v1.raw_ops.LSTMBlockCell

tf.raw_ops.LSTMBlockCell(
    x, cs_prev, h_prev, w, wci, wcf, wco, b, forget_bias=1, cell_clip=3,
    use_peephole=False, name=None
)

This implementation uses 1 weight matrix and 1 bias vector, and there's an optional peephole connection.

This kernel op implements the following mathematical equations:

xh = [x, h_prev]
[i, f, ci, o] = xh * w + b
f = f + forget_bias

if not use_peephole:
  wci = wcf = wco = 0

i = sigmoid(cs_prev * wci + i)
f = sigmoid(cs_prev * wcf + f)
ci = tanh(ci)

cs = ci .* i + cs_prev .* f
cs = clip(cs, cell_clip)

o = sigmoid(cs * wco + o)
co = tanh(cs)
h = co .* o

Args
`x`	A `Tensor`. Must be one of the following types: `half`, `float32`. The input to the LSTM cell, shape (batch_size, num_inputs).
`cs_prev`	A `Tensor`. Must have the same type as `x`. Value of the cell state at previous time step.
`h_prev`	A `Tensor`. Must have the same type as `x`. Output of the previous cell at previous time step.
`w`	A `Tensor`. Must have the same type as `x`. The weight matrix.
`wci`	A `Tensor`. Must have the same type as `x`. The weight matrix for input gate peephole connection.
`wcf`	A `Tensor`. Must have the same type as `x`. The weight matrix for forget gate peephole connection.
`wco`	A `Tensor`. Must have the same type as `x`. The weight matrix for output gate peephole connection.
`b`	A `Tensor`. Must have the same type as `x`. The bias vector.
`forget_bias`	An optional `float`. Defaults to `1`. The forget gate bias.
`cell_clip`	An optional `float`. Defaults to `3`. Value to clip the 'cs' value to.
`use_peephole`	An optional `bool`. Defaults to `False`. Whether to use peephole weights.
`name`	A name for the operation (optional).

Returns
A tuple of `Tensor` objects (i, cs, f, o, ci, co, h).
`i`	A `Tensor`. Has the same type as `x`.
`cs`	A `Tensor`. Has the same type as `x`.
`f`	A `Tensor`. Has the same type as `x`.
`o`	A `Tensor`. Has the same type as `x`.
`ci`	A `Tensor`. Has the same type as `x`.
`co`	A `Tensor`. Has the same type as `x`.
`h`	A `Tensor`. Has the same type as `x`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/LSTMBlockCell

Docs

Docs4dev

Title here

tf.raw_ops.LSTMBlockCell

View aliases