mindspore构建网络时，自己定义的函数或层不能正常使用

问题描述：

构建网络时，使用了自己定义的一个功能函数

 

大致功能是将fm转换到x1的大小

在网络中直接使用这个函数，构建网络时产生如下错误：

 

怀疑是需要使用类似层的定义，所以将函数改为类后如下

 

仍然报错，错误如下：

 

解答：

是不是没有导入Rasize()，可以参考这样实现：

from mindspore.dataset.vision.c_transforms import Resize

import numpy as np

    

x1 = np.ones((8,3,256,256))

fm = np.ones((512,512))

resize = Resize((x1.shape[2], x1.shape[3]))

fm = resize(fm)

print(fm.shape)

mindspore.dataset.vision.c_transforms.Resize — MindSpore master documentation