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.

参数

• 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.

参数

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.

参数

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.

参数

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.

参数

activation_type – [in] Define the activation type.

bool get_has_bias() const

Method to get has_bias attribute.

返回

a boolean value

int64_t get_axis() const

Method to get axis attribute.

返回

axis

bool get_use_axis() const

Method to get use_axis attribute.

返回

a boolean value.

ActivationType get_activation_type() const

Method to get activation type.

返回

activation type.