tf.keras.layers.LSTM/tf.keras.layers.LSTMCell

简介: 【8月更文挑战第20天】tf.keras.layers.LSTM/tf.keras.layers.LSTMCell。

tf.keras.layers.LSTM/tf.keras.layers.LSTMCell
tf.keras.layers.LSTM/tf.keras.layers.LSTMCell可配置的参数,主要有:
units:输出维度;
input_shape (timestep, input_dim),timestep可以设置为None,input_dim为输入数据维度;
activation:激活函数;
recurrent_activation: 为循环步施加的激活函数;
return_sequences:=True时,返回全部序列;=False时,返回输出序列中的最后一个cell的输出;
return_state: 布尔值。除了输出之外是否返回最后一个状态;
dropout:0~1之间的浮点数,控制输入线性变换的神经元断开比例;
recurrent_dropout:0~1之间的浮点数,控制循环状态的线性变换的神经元断开比例。

代码:
import numpy as np
inputs = tf.keras.Input(shape=(3, 1))
lstm = layers.LSTM(1, return_sequences=True)(inputs)
model_lstm_1 = tf.keras.models.Model(inputs=inputs, outputs=lstm)

inputs = tf.keras.Input(shape=(3, 1))
lstm = layers.LSTM(1, return_sequences=False)(inputs)
model_lstm_2 = tf.keras.models.Model(inputs=inputs, outputs=lstm)

t1, t2, t3 序列

data = [[[0.1],
[0.2],
[0.3]]]
print(data)
print("当return_sequences=True时的输出",model_lstm_1.predict(data))
print("当return_sequences=False时的输出",model_lstm_2.predict(data))

输出:
[[[0.1], [0.2], [0.3]]]
当return_sequences=True时的输出 [[[-0.0106758 ]
[-0.02711176]
[-0.04583194]]]
当return_sequences=False时的输出 [[0.05914127]]

LSTMcell是LSTM层的实现单元。
LSTM是一个LSTM网络层
LSTMCell是一个单步的计算单元,即一个LSTM UNIT。

LSTM

tf.keras.layers.LSTM(16, return_sequences=True)

LSTMCell

x = tf.keras.Input((None, 3))
y = layers.RNN(layers.LSTMCell(16))(x)
model_lstm_3= tf.keras.Model(x, y)

目录
相关文章
|
2月前
|
机器学习/深度学习 TensorFlow API
Keras介绍
【7月更文挑战第27天】Keras介绍。
36 2
|
23天前
tf.keras.layers.Dense
【8月更文挑战第20天】tf.keras.layers.Dense。
30 2
|
23天前
|
算法框架/工具 数据格式
tf.keras.layers.Conv2D
【8月更文挑战第20天】tf.keras.layers.Conv2D。
28 2
|
24天前
tf.keras.layers.MaxPooling2D/AveragePooling2D
【8月更文挑战第19天】tf.keras.layers.MaxPooling2D/AveragePooling2D。
23 0
|
23天前
|
机器学习/深度学习
网络层构建(tf.keras.layers)
【8月更文挑战第20天】网络层构建(tf.keras.layers)。
26 2
|
1月前
|
机器学习/深度学习 TensorFlow 算法框架/工具
【Tensorflow+Keras】tf.keras.layers.Bidirectional()的解析与使用
本文解析了TensorFlow和Keras中的`tf.keras.layers.Bidirectional()`层,它用于实现双向RNN(如LSTM、GRU)的神经网络结构。文章详细介绍了该层的参数配置,并通过实例演示了如何构建含有双向LSTM层的模型,以及如何使用IMDB数据集进行模型训练和评估。
41 8
|
4月前
|
机器学习/深度学习 TensorFlow API
Keras 的模型(Model)和层(Layers)的介绍
Keras 的模型(Model)和层(Layers)的介绍
94 1
|
机器学习/深度学习 PyTorch 算法框架/工具
|
PyTorch 算法框架/工具
PyTorch的nn.Linear()详解
从输入输出的张量的shape角度来理解,相当于一个输入为[batch_size, in_features]的张量变换成了[batch_size, out_features]的输出张量。
447 0