关于向量拼接和nn.SequentialCell

小乐快乐

于 2024-06-28 16:04:35 发布

阅读量215

点赞数 2

文章标签： python 深度学习机器学习

本文链接：https://blog.csdn.net/weixin_45666880/article/details/139196542

版权

问题描述

我在Unet的转置卷积以后需要对两个路径上的向量按照通道进行拼接，ops.concat不是cell的子类，是不是只能一个一个的写了？

解答方案

SequentialCell里只能拼接nn下的接口，或者自定义的Cell，写法如：

self.double_conv = nn.SequentialCell(
            [nn.Conv2d(in_channels, mid_channels, kernel_size=3, has_bias=True,
                       weight_init=init_value_0, bias_init="zeros", pad_mode="valid"),
             nn.ReLU(),
             nn.Conv2d(mid_channels, out_channels, kernel_size=3, has_bias=True,
                       weight_init=init_value_1, bias_init="zeros", pad_mode="valid"),
             nn.ReLU()]
        )

concat一般是拼接Tensor，用法可参考：

data1 = Tensor(np.array([[0, 1], [2, 3]]).astype(np.float32))
data2 = Tensor(np.array([[4, 5], [6, 7]]).astype(np.float32))
output = ops.concat((data1, data2), axis=0)

print(output)
print("shape:\n", output.shape)