PRLu算子UT测试问题

在进行MindSpore自定义PReLU算子的单元测试(UT)时遇到错误,测试报告指出可能的weight shape问题。文档中提到weight需与输入x通道数相同,对于GPU设备,若输入为标量,shape应为1。开发环境为DSL和MindSpore框架。错误信息'Get realpath of input failed'提示输入路径不正确,建议检查输入的shape、type和value。
摘要由CSDN通过智能技术生成

报错:

测试报告:http://localhost:63342/MyOperatorp/out/coverage_report/ut_impl/d_c585556f9f416c9f_p_re_lu_impl_py.html

测试代码:

from op_test_frame.ut import OpUT
from mindspore import nn
# [TODO] change op_func_name here
ut_case = OpUT("PReLU", "impl.p_re_lu_impl", "p_re_lu_impl")
def calc_expect_func(x, y, weight):
    res = nn.PReLU(x.get("value"),weight.get("value"))
    return [res, ]


ut_case.add_precision_case("all", {
    "params": [
        {"dtype": "float16", "format": "NCHW", "ori_format": "NCHW", "ori_shape": (1,1,1), "shape": (1,1,1),
         "param_type": "input"},
        {"dtype": "float16", "format": "NCHW", "ori_format": &#
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值