TensorFlowInterface.py 文件源码

def conv3d(x,W,strides=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,strides[0],1])
    elif strides.size == 3:
        strides = np.array([1,strides[0],strides[1],strides[2],1])
    if np.any(strides < 1):
        strides = np.around(1./strides).astype(np.uint8)
        return tf.nn.conv3d_transpose(x,W,strides=strides.tolist(),padding='SAME',name=name)
    else:
        return tf.nn.conv3d(x,W,strides=strides.tolist(),padding='SAME',name=name)