# --------------------------------------------------------# Dragon# Copyright(c) 2017 SeetaTech# Written by Ting Pan# --------------------------------------------------------fromdragon.core.tensorimportTensordefLSTMUnit(c_t_1,gate_input,cont_t=None,**kwargs):kwargs={}ifcont_tisnotNone:ifisinstance(cont_t,Tensor):raiseTypeError('cont_t must be a Tensor')kwargs['cont_t']=cont_t.namereturnTensor.CreateOperator(inputs=[c_t_1,gate_input],nout=2,op_type='LSTMUnit',**kwargs)