Class Conv2D

Defined in File conv2d.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Derived Type

public mindspore::ops::Conv2DFusion (Class Conv2DFusion)

Class Documentation

class Conv2D : public mindspore::ops::BaseOperator 

2D convolution layer. Refer to Python API mindspore.ops.Conv2D for more details.

Subclassed by mindspore::ops::Conv2DFusion

Public Functions

inline Conv2D(): Constructor.

void Init(int64_t out_channel, const std::vector<int64_t> &kernel_size, int64_t mode = 1, const PadMode &pad_mode = VALID, const std::vector<int64_t> &pad = {0, 0, 0, 0}, const std::vector<int64_t> &stride = {1, 1, 1, 1}, const std::vector<int64_t> &dilation = {1, 1, 1, 1}, int64_t group = 1, const Format &format = NCHW): Init. Refer to the parameters of Python API mindspore.ops.Conv2D for the inputs.

void set_kernel_size(const std::vector<int64_t> &kernel_size): Set kernel_size.

void set_stride(const std::vector<int64_t> &stride): Set stride.

void set_dilation(const std::vector<int64_t> &dilation): Set dilation.

void set_pad_mode(const PadMode &pad_mode): Set pad_mode.

void set_pad(const std::vector<int64_t> &pad): Set pad.

void set_mode(int64_t mode): Set mode.

void set_group(int64_t group): Set group.

void set_out_channel(int64_t out_channel): Set out_channel.

void set_format(const Format &format): Set format.

std::vector<int64_t> get_kernel_size() const

Get kernel_size.

返回: kernel_size.

std::vector<int64_t> get_stride() const

Get stride.

返回: stride.

std::vector<int64_t> get_dilation() const

Get dilation.

返回: dilation.

PadMode get_pad_mode() const

Get pad_mode.

返回: pad_mode.

std::vector<int64_t> get_pad() const

Get pad.

返回: pad.

int64_t get_mode() const

Get mode.

返回: mode.

int64_t get_group() const

Get group.

返回: group.

int64_t get_out_channel() const

Get out_channel.

返回: out_channel.

Format get_format() const

Get format.

返回: format.