135a56e0935fbb04811f8ce7b9f514f498212f71,niftynet/layer/crf.py,,ftheta,#Any#Any#Any#Any#Any#Any#,148

Before Change

    // Message Passing
    data = tf.reshape(tf.nn.softmax(H1), [batch_size, -1, n_ch])
    Q1 = [None] * len(permutohedrals)
    with tf.device("/cpu:0"):
        for idx, permutohedral in enumerate(permutohedrals):
            Q1[idx] = tf.reshape(
                _permutohedral_gen(permutohedral, data, name + str(idx)),
                U.shape)

After Change

    for idx, permutohedral in enumerate(permutohedrals):
        // Message Passing
        Q = _permutohedral_gen(permutohedral, H1, name + str(idx))
        Q.set_shape([n_voxels, n_ch])
        // Weighting Filtered Outputs
        Q1 = Q1 + Q * kernel_weights[idx]

    // Compatibility Transform, Adding Unary Potentials

Instances