Class GRU
Defined in File gru.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class GRU : public mindspore::ops::BaseOperator
GRU defined the GRU operator prototype.
Public Functions
-
inline GRU()
Constructor.
-
void Init(bool bidirectional = false)
Method to init the op’s attributes.
- Parameters
bidirectional – [in] Define a boolean value to indicate whether the gru is single or double direction.
-
void set_bidirectional(bool bidirectional)
Method to set bidirectional attribute.
- Parameters
bidirectional – Define a boolean value to indicate whether the gru is single or double direction.
-
bool get_bidirectional() const
Method to get bidirectional attribute.
- Returns
a boolean value.
-
inline GRU()