Class ConstantOfShape

• Defined in File constant_of_shape.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class ConstantOfShape : public mindspore::ops::BaseOperator

ConstantOfShape defined ConstantOfShape operator prototype of lite.

Public Functions

inline ConstantOfShape()

Constructor.

void Init(int64_t data_type, const std::vector<float> &value)

Method to init the op’s attributes.

Parameters

• data_type – [in] Define data type of output data.

• value – [in] Define value of output data.

void set_data_type(int64_t data_type)

Method to set data type attribute.

Parameters

data_type – [in] Define data type of output data.

void set_value(const std::vector<float> &value)

Method to set value attribute.

Parameters

value – [in] Define value of output data.

int64_t get_data_type() const

Method to get data type attribute.

Returns

data type attribute.

std::vector<float> get_value() const

Method to get value attribute.

Returns

value attribute.