Class ResizeNearestNeighbor

• Defined in File resize_nearest_neighbor.h

Inheritance Relationships

Base Type

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

Class Documentation

class ResizeNearestNeighbor : public mindspore::ops::PrimitiveC

Resizes the input tensor by using the nearest neighbor algorithm. Refer to Python API mindspore.ops.ResizeNearestNeighbor for more details.

Public Functions

inline ResizeNearestNeighbor()

Constructor.

~ResizeNearestNeighbor() = default

Destructor.

void Init(const std::vector<int64_t> &size, const bool align_corners = false)

Init. Refer to the parameters of Python API mindspore.ops.ResizeNearestNeighbor for the inputs.

void set_size(const std::vector<int64_t> &size)

Set size.

void set_align_corners(const bool align_corners)

Set align_corners.

std::vector<int64_t> get_size() const

Get size.

Returns

size.

bool get_align_corners() const

Get align_corners.

Returns

align_corners.