tensorflow2.0 中lstm的实现

tensorflow2.0 中lstm的实现

tensorflow2.0 中lstm的实现

from __future__ import absolute_import, division, print_function, unicode_literals

import collections
import matplotlib.pyplot as plt
import numpy as np

try:
  # %tensorflow_version only exists in Colab.
  %tensorflow_version 2.x
except Exception:
  pass
import tensorflow as tf

from tensorflow.keras import layers

设计模型

batch_size = 128   # 大小可变，一次处理多少个样本
# Each MNIST image batch is a tensor of shape (batch_size, 28, 28).
# Each input sequence will be of size (28, 28) (height is treated like time).
input_dim = 28  # 必须和特征的列数目相等

units = 64   #  LSTM 中
output_size = 10  # labels are from 0 to 9

Build the RNN model

def build_model(allow_cudnn_kernel=True):
  # CuDNN is only available at the layer level, and not at the cell level.
  #