关于正向卷积运算的实现代码求助

"""
多通道卷积前向过程
:param inputs: 卷积层矩阵,形状(N,H,W,C),N为batch_size,C为通道数
:param kernel: 卷积核,形状(k,k,C,D), C为输入通道数,D为输出通道数
:param bias: 偏置,形状(D,)
:param padding: padding
:param strides: 步长
:return: 卷积结果outputs
"""
def conv2d_forward(inputs, kernel, bias, padding=1, strides=1):
    outputs = []
    N, H, W, C = inputs.shape
    k, k, _, D = kernel.shape

    # 卷积实现

    return outputs

求正向卷积运算的实现代码。没用过pytorch,有哪位大神可以帮忙看看吗?可以给出实现代码吗?感激不尽,尝试了很久没写出来。tensorflow、python都可以。

跪求!!!

麻烦在这个代码基础上实现。

#Python##学习路径#
全部评论
希望能连同测试代码一起给出来,感激不尽
点赞 回复
分享
发布于 2021-05-06 21:41
可以去搜一下img2col算法,是torch和caffe里的卷积实现方式好像。
点赞 回复
分享
发布于 2021-05-19 17:36
联想
校招火热招聘中
官网直投

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务