tensorflow.nn.conv2d等效的py代码

最新推荐文章于 2024-06-29 22:33:22 发布

pialone

最新推荐文章于 2024-06-29 22:33:22 发布

阅读量203

点赞数 1

分类专栏： python tensorflow 文章标签： tensorflow 深度学习人工智能

本文链接：https://blog.csdn.net/pialone/article/details/121784082

版权

本文通过Python代码模拟tensorflow.nn.conv2d的核心计算过程，解释输入、卷积核、滑动步长和填充方式等关键参数的作用，帮助读者深入理解卷积操作。

摘要由CSDN通过智能技术生成

为了能理解conv2d函数，最好的学习就是自己再制造它，那就用代码模拟conv2d里体是怎么计算的。

import math

def conv2d(input, filter, **kwargs):
    strides= 'strides' in kwargs and kwargs['strides'] or [1,1,1,1]
    padding = 'padding' in kwargs and kwargs['padding'] or 'SAME'
    assert(padding =='SAME' or  padding =='VALID')
    assert(strides[0]==1 and strides[3]==1)

    in_batch  =len(input)
    in_height =len(input[0])
    in_width  =len(input[0][0])
    in_channel=len(input[0][0][0])
    
    fi_height  = len(filter)
    fi_width   = len(filter[0])
    assert(in_channel==len(filter[0][0]))
    out_channel = len(filter[0][0][0])
    
    if padding == "VALID":
       assert(in_height >= fi_height)
       assert(in_width  >= fi_width)

    def take_sample(T, i, j):
        M = []
        for y in range(f