rvc-project / obs-rvc Goto Github PK

    let conv_input_size = buffer_frame_size + search_frame_size;
    let conv_input = input_buffer
        .slice(s![..conv_input_size])
        .into_shape((1, 1, conv_input_size))?;
    let sola_buffer_view = sola_buffer
        .into_shape((1, 1, buffer_frame_size))?;

    let cor_nom = conv_input.conv_fft(
        &sola_buffer_view, ndarray_conv::ConvMode::Valid, 
        ndarray_conv::PaddingMode::Zeros
    )?;

    let cor_den_filler = ndarray::Array3::<f32>::ones((1, 1, buffer_frame_size));
    let mut cor_den = conv_input.mapv(|x| x.powi(2)).conv_fft(
        &cor_den_filler, ndarray_conv::ConvMode::Valid, 
        ndarray_conv::PaddingMode::Zeros
    )?;