ModifiedNeuralNet.py 文件源码

def process_pre_post_w(padding_arr, zeros_arr):
    argmax = T.argmax(padding_arr)
    zeros_arr = ifelse(T.eq(padding_arr[argmax], 0), zeros_arr,
                       T.set_subtensor(zeros_arr[argmax-2:argmax+3], 1.5 / (T.sum(padding_arr[argmax-2:argmax+3]))))
    return_arr = (zeros_arr * padding_arr)[2: -2]
    return return_arr