Class Pad

Defined in File pad.h

Inheritance Relationships

Base Type

public mindspore::ops::PrimitiveC (Class PrimitiveC)

Derived Type

public mindspore::ops::PadFusion (Class PadFusion)

Class Documentation

class Pad : public mindspore::ops::PrimitiveC

Pads the input tensor according to the paddings. Refer to Python API mindspore.ops.Pad for more details.

Subclassed by mindspore::ops::PadFusion

Public Functions

inline Pad(): Constructor.

~Pad() = default: Destructor.

void Init(const std::vector<std::vector<int64_t>> &paddings): Init. Refer to the parameters of Python API mindspore.ops.Pad for the inputs.

void set_paddings(const std::vector<std::vector<int64_t>> &paddings): Set paddings.

std::vector<std::vector<int64_t>> get_paddings() const

Get paddings.

Returns: paddings.