|
|
|
@ -1790,7 +1790,7 @@ class ScatterNd(PrimitiveWithInfer):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class ResizeNearestNeighbor(PrimitiveWithInfer):
|
|
|
|
|
"""
|
|
|
|
|
r"""
|
|
|
|
|
Resize the input tensor by using nearest neighbor algorithm.
|
|
|
|
|
|
|
|
|
|
Resize input tensor to given size by using nearest neighbor algorithm. The nearest
|
|
|
|
@ -1806,7 +1806,7 @@ class ResizeNearestNeighbor(PrimitiveWithInfer):
|
|
|
|
|
- **input_x** (Tensor) - The input tensor. The shape of the tensor is :math:`(N, C, H, W)`.
|
|
|
|
|
|
|
|
|
|
Outputs:
|
|
|
|
|
Tensor, the shape of the output tensor is :math:`(N, NEW_C, NEW_H, W)`.
|
|
|
|
|
Tensor, the shape of the output tensor is :math:`(N, NEW\_C, NEW\_H, W)`.
|
|
|
|
|
|
|
|
|
|
Examples:
|
|
|
|
|
>>> input_tensor = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32)
|
|
|
|
|