| 2418 | // ggml_mean |
| 2419 | |
| 2420 | struct ggml_tensor * ggml_mean( |
| 2421 | struct ggml_context * ctx, |
| 2422 | struct ggml_tensor * a) { |
| 2423 | int64_t ne[4] = { 1, a->ne[1], a->ne[2], a->ne[3] }; |
| 2424 | struct ggml_tensor * result = ggml_new_tensor(ctx, GGML_TYPE_F32, 4, ne); |
| 2425 | |
| 2426 | result->op = GGML_OP_MEAN; |
| 2427 | result->src[0] = a; |
| 2428 | |
| 2429 | return result; |
| 2430 | } |
| 2431 | |
| 2432 | // ggml_argmax |
| 2433 |