| 2487 | // ggml_mean |
| 2488 | |
| 2489 | struct ggml_tensor * ggml_mean( |
| 2490 | struct ggml_context * ctx, |
| 2491 | struct ggml_tensor * a) { |
| 2492 | int64_t ne[4] = { 1, a->ne[1], a->ne[2], a->ne[3] }; |
| 2493 | struct ggml_tensor * result = ggml_new_tensor(ctx, GGML_TYPE_F32, 4, ne); |
| 2494 | |
| 2495 | result->op = GGML_OP_MEAN; |
| 2496 | result->src[0] = a; |
| 2497 | |
| 2498 | return result; |
| 2499 | } |
| 2500 | |
| 2501 | // ggml_argmax |
| 2502 |