当前位置:网站首页>初始化层实现

初始化层实现

2022-06-23 06:18:00 休斯顿凤梨

应用:

卷积后不改变原图(H,W)

功能:

 

1.设置location

2.设定指定value

3.设定padding

实现:

#Refer to the official website
import paddle.fluid as fluid
from paddle.fluid.dygraph import Conv2D
from paddle.fluid.initializer import NumpyArrayInitializer
import numpy as np 

np.set_printoptions(precision=1)
def main():
    x = np.random.rand(6,6).astype(np.float32)
    print(x.shape)
    print(x)
    x = x[np.newaxis,np.newaxis,:,:]

    d = 2
    with fluid.dygraph.guard(fluid.CPUPlace()):
        w = np.ones((1,1,3,3)).astype(np.float32)
        pa = fluid.ParamAttr(name = 'conv',initializer=NumpyArrayInitializer(w))
        dilated_conv = Conv2D(num_channels=1,
                              num_filters=1,
                              filter_size=3,
                              dilation=d,
                              padding=0,
                              stride=1,
                              param_attr=pa)
        print(dilated_conv.weight.numpy().squeeze((0,1)))
        print(f'dilation={d}')
        x = fluid.dygraph.to_variable(x)
        y = dilated_conv(x)
        y = y.numpy().squeeze((0,1))
        print(y.shape)
        print(y)

                             
    

运行:

原网站

版权声明
本文为[休斯顿凤梨]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_45779396/article/details/124968853