Tensorflow常见模型实现之一（LSTM/BiLSTM）

最新推荐文章于 2024-08-07 15:23:57 发布

zchenack

最新推荐文章于 2024-08-07 15:23:57 发布

阅读量8.3k

点赞数 2

分类专栏：自然语言处理

本文链接：https://blog.csdn.net/hustchenze/article/details/83041882

版权

1. LSTM

import tensorflow as tf
import tensorflow.contrib as contrib
from tensorflow.python.ops import array_ops

class lstm(object):
    def __init__(self, in_data, hidden_dim, batch_seqlen=None, flag='concat'):
        self.in_data = in_data
        self.hidden_dim = hidden_dim
        self.batch_seqlen = batch_seqlen
        self.flag = flag
        
        lstm_cell = contrib.rnn.LSTMCell(self.hidden_dim)
        out, _ = tf.nn.dynamic_rnn(cell=lstm_cell, inputs=self.in_data, sequence_length=self.batch_seqlen,dtype=tf.float32)

        if flag=='all_ht':
            self.out = out
        if flag = 'first_ht':
            self.out = out[:,0,:]
        if flag = 'last_ht':
            self.out = out[:,-1,:]
        if flag = 'concat':
            self.out = tf.concat([out[:,0,:], out[:,-1,:]],1)</