---
title: LSTM
description: "- x:"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/aolapi/operatorlist_00126.html
sourcePath: /source/zh/canncommercial/900/API/aolapi/operatorlist_00126.html
indexId: 1660a2c40d5e9eb1947d7006e720a49d2b7ccc3b6d4d4b6a8e512a38ebc22b1564
---
# LSTM

#### 输入

- x:

  - 是否必填: 必填
  - 数据类型: float16
  - 参数解释: 时间变化数据x，[T,N,input_size]
  - 规格限制: T最大支持256
- cont:

  - 是否必填: 必填
  - 数据类型: float16、float32
  - 参数解释: 序列连续性标记，[T,N]
  - 规格限制: T最大支持256
- w_x:

  - 是否必填: 必填
  - 数据类型: float16
  - 参数解释: x权重，维度为[input_size, 4×num_output]
  - 规格限制: 无
- bias:

  - 是否必填: 必填
  - 数据类型: float16、float32
  - 参数解释: 偏置，维度为[4×num_output]
  - 规格限制: 无
- w_h:

  - 是否必填: 必填
  - 数据类型: float16
  - 参数解释: h权重，维度为[4×num_output, num_output]
  - 规格限制: 无
- x_static:

  - 是否必填: 非必填
  - 数据类型: float16
  - 参数解释: 时间不变的静态数据x_static[N,input_size]；接口当前保留
  - 规格限制: 当传入static输入时，要求inputFM和HiddenSize都必须是16的倍数
- h_0:

  - 是否必填: 非必填
  - 数据类型: float16、float32
  - 参数解释: expose_hidden==True时输入，初始隐藏状态[1,N,num_output]
  - 规格限制: 无
- c_0:

  - 是否必填: 非必填
  - 数据类型: float16、float32
  - 参数解释: expose_hidden==True时输入，初始cell状态 [1,N,num_output]
  - 规格限制: 无
- w_x_static:

  - 是否必填: 非必填
  - 数据类型: float16
  - 参数解释: x_static权重，维度为[4×num_output,input_size]; 接口当前保留
  - 规格限制: 无


#### 属性

- num_output:

  - 是否必填: 必填
  - 数据类型: uint
  - 参数解释: 输出特征数，必须大于0
  - 规格限制: 必须大于0
- expose_hidden:

  - 是否必填: 非必填
  - 数据类型: bool
  - 参数解释: 是否有初始隐藏状态和cell初始状态，默认为False
  - 规格限制: 无


#### 输出

- h:

  - 是否必填: 必填
  - 数据类型: float16、float32
  - 参数解释: 输出的tensor (T×N×num_output)
  - 规格限制: 无

- h_t:

  - 是否必填: 非必填
  - 数据类型: float16、float32
  - 参数解释: 最后timestep的隐藏状态 (1×N×num_output)
  - 规格限制: 无

- c_t:

  - 是否必填: 非必填
  - 数据类型: float16、float32
  - 参数解释: 最后timestep的cell状态 (1×N×num_output)
  - 规格限制: 无
