Class FullConnection

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.