Function view_batch_matrix

src/models/ace_step/vae_decoder.cpp:180–194

        frames,
        channels,
        input.tensor->nb[1],
        static_cast<size_t>(batch_index) * input.tensor->nb[2]);
    return core::wrap_tensor(view, core::TensorShape::from_dims({channels, frames}), input.type);
}

core::TensorValue build_snake1d_exact_bct(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const SnakeExactWeights & weights,
    int64_t channels) {
    const auto input_f32 = ensure_f32(ctx, ensure_contiguous_nontransposed(ctx, input));
    auto alpha = ensure_f32(ctx, weights.alpha);
    auto beta_inv = ensure_f32(ctx, weights.beta_inv);
    if (alpha.shape.rank != 3 || alpha.shape.dims[0] != 1 || alpha.shape.dims[1] != channels || alpha.shape.dims[2] != 1 ||
        beta_inv.shape.rank != 3 || beta_inv.shape.dims[0] != 1 || beta_inv.shape.dims[1] != channels || beta_inv.shape.dims[2] != 1) {
        throw std::runtime_error("ACE-Step VAE Snake1d weight shape mismatch");
    }
    if (input_f32.shape.rank != 3) {