Class Conv2DTranspose

Defined in File conv2d_transpose.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Derived Type

public mindspore::ops::Conv2dTransposeFusion (Class Conv2dTransposeFusion)

Class Documentation

class Conv2DTranspose : public mindspore::ops::BaseOperator 

2D transposed convolution layer. Refer to Python API mindspore.nn.Conv2dTranspose for more details.

Subclassed by mindspore::ops::Conv2dTransposeFusion

Public Functions

inline Conv2DTranspose(): Constructor.

void Init(int64_t in_channel, 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}, const std::vector<int64_t> &dilation = {1, 1}, int64_t group = 1, const Format &format = NCHW, const std::vector<int64_t> &pad_list = {0, 0, 0, 0}): Init. Refer to the parameters of Python API mindspore.nn.Conv2dTranspose for the inputs.

void set_in_channel(int64_t in_channel): Set in_channel.

void set_out_channel(int64_t out_channel): Set out_channel.

virtual 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.

virtual 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_format(const Format &format): Set format.

void set_pad_list(const std::vector<int64_t> &pad_list): Set pad_list.

int64_t get_in_channel() const

Get in_channel.

返回: in_channel.

int64_t get_out_channel() const

Get out_channel.

返回: out_channel.

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.

Format get_format() const

Get format.

返回: format.

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

Get pad_list.

返回: pad_list.