Class ConstantOfShape

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.