torch_npu.npu_gru(input, hx, weight_input, weight_hidden, bias_input, bias_hidden, seq_length, has_biases, num_layers, dropout, train, bidirectional, batch_first) -> (Tensor, Tensor, Tensor, Tensor, Tensor, Tensor)
计算DynamicGRUV2。
- 参数解释：
- input (Tensor) - 数据类型：float16；格式：FRACTAL_NZ。
- hx (Tensor) - 数据类型：float16, float32；格式：FRACTAL_NZ。
- weight_input (Tensor) - 数据类型：float16；格式：FRACTAL_Z。
- weight_hidden (Tensor) - 数据类型：float16；格式：FRACTAL_Z。
- bias_input (Tensor) - 数据类型：float16, float32；格式：ND。
- bias_hidden (Tensor) - 数据类型：float16, float32；格式：ND。
- seq_length (Tensor) - 数据类型：int32；格式：ND。
- has_biases (Bool，默认值为True)
- num_layers (Int)
- dropout (Float)
- train (Bool，默认值为True) - 标识训练是否在op进行的bool参数。
- bidirectional (Bool，默认值为True)
- batch_first (Bool，默认值为True)
- Returns:
- y (Tensor) - 数据类型：float16, float32；格式：FRACTAL_NZ。
- output_h (Tensor) - 数据类型：float16, float32；格式：FRACTAL_NZ。
- update (Tensor) - 数据类型：float16, float32；格式：FRACTAL_NZ。
- reset (Tensor) - 数据类型：float16, float32；格式：FRACTAL_NZ。
- new (Tensor) - 数据类型：float16, float32；格式：FRACTAL_NZ。
- hidden_new (Tensor) - 数据类型：float16, float32；格式：FRACTAL_NZ。
- 约束条件：
无
- 示例：
无