Class GRU

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.