Environment

• Hardware Environment(Ascend/GPU/CPU):

Uncomment only one /device <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/device cpu

• Software Environment:
  -- MindSpore version (source or binary):
  -- Python version (e.g., Python 3.7.5):
  -- OS platform and distribution (e.g., Linux Ubuntu 16.04):
  -- GCC/Compiler version (if compiled from source):

Related testcase

Steps to reproduce the issue

1. issue source link: https://bbs.huaweicloud.com/forum/thread-85925-1-1.html

Describe the current behavior

import mindspore
from mindspore import Tensor, context
import numpy as np
import mindspore.nn as nn

context.set_context(mode=context.GRAPH_MODE,device_target="CPU")

input_x = Tensor(np.array([-1, 2, -3, 2, -1]), mindspore.float16)
relu = nn.ReLU()
relu(input_x)

error info:

RuntimeError: mindspore\ccsrc\backend\kernel_compiler\cpu\mkldnn\relu_cpu_kernel.cc:28 InitKernel] relu kernel dims invalid 1

if use data [[-1, 2, -3, 2, -1]] instead [-1, 2, -3, 2, -1] it will run success.

Describe the expected behavior

Related log / screenshot

Special notes for this issue