Class FullConnection
Defined in File full_connection.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class FullConnection : public mindspore::ops::BaseOperator
FullConnection defined FullConnection operator prototype of lite.
Public Functions
-
inline FullConnection()
Constructor.
-
void Init(const bool has_bias, const int64_t axis, const bool use_axis, const ActivationType &activation_type)
Method to init the op’s attributes.
- Parameters
has_bias – [in] Define a boolean value to indicate the op has bias or not.
axis – [in] Define a axis that the inner product is done along with
use_axis – [in] Define a boolean value to indicate the op uses a axis or not.
activation_type – [in] Define the activation type.
-
void set_has_bias(const bool has_bias)
Method to set has_axis attribute.
- Parameters
has_bias – [in] Define a boolean value to indicate the op has bias or not.
-
void set_axis(const int64_t axis)
Method to set axis attribute.
- Parameters
axis – [in] Define a axis the inner product must be along with
-
void set_use_axis(const bool use_axis)
Method to set use_axis attribute.
- Parameters
use_axis – [in] Define a boolean value to indicate the op uses a axis or not.
-
void set_activation_type(const ActivationType &activation_type)
Method to set activation type.
- Parameters
activation_type – [in] Define the activation type.
-
bool get_has_bias() const
Method to get has_bias attribute.
- Returns
a boolean value
-
int64_t get_axis() const
Method to get axis attribute.
- Returns
axis
-
bool get_use_axis() const
Method to get use_axis attribute.
- Returns
a boolean value.
-
ActivationType get_activation_type() const
Method to get activation type.
- Returns
activation type.
-
inline FullConnection()