TensorFlowInterface.py 文件源码

python
阅读 29 收藏 0 点赞 0 评论 0

项目:IntroToDeepLearning 作者: robb-brown 项目源码 文件源码
def conv2d(x,W,strides=[1,1,1,1],name=None):
    # return an op that convolves x with W
    strides = np.array(strides)
    if strides.size == 1:
        strides = np.array([1,strides,strides,1])
    elif strides.size == 2:
        strides = np.array([1,strides[0],strides[1],1])
    if np.any(strides < 1):
        strides = np.around(1./strides).astype(np.uint8)
        return tf.nn.conv2d_transpose(x,W,strides=strides.tolist(),padding='SAME',name=name)
    else:
        return tf.nn.conv2d(x,W,strides=strides.tolist(),padding='SAME',name=name)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号