Function reshape_codebook_indices

src/models/heartmula/codec.cpp:605–615

    const HeartCodecResidualUnitWeights & weights,
    int64_t channels,
    int64_t kernel_size,
    int64_t dilation) {
    auto x = causal_conv1d(ctx, input, weights.conv1, channels, channels, kernel_size, dilation);
    x = prelu(ctx, x, weights.activation1);
    x = causal_conv1d(ctx, x, weights.conv2, channels, channels, 1);
    x = prelu(ctx, x, weights.activation2);
    return modules::AddModule{}.build(ctx, input, x);
}

core::TensorValue scalar_decoder(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & latents,
    const HeartCodecScalarDecoderWeights & weights,
    const HeartCodecConfig & config) {