4c9acddb0bfbbd66d8ab26687bb0079630d5f622,niftynet/layer/crf.py,,ftheta,#,163

Before Change

    :return: updated mean-field distribution
    
    batch_size, n_ch = U.shape.as_list()[0], U.shape.as_list()[-1]
    n_voxels = np.prod(U.shape.as_list()[:-1])

    H1 = tf.reshape(tf.nn.softmax(H1), [batch_size, -1, n_ch])
    Q1 = 0

After Change

    :param name: layer name
    :return: updated mean-field distribution
    
    batch_size, n_voxels, n_ch = U.shape.as_list()
    H1 = tf.nn.softmax(H1)
    Q1 = 0
    for idx, permutohedral in enumerate(permutohedrals):
        // Message Passing

Instances