TensorFlowInterface.py 文件源码

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)