Skip to content

Commit ba6f469

Browse files
committed
better representation of max,min,sum and mean
1 parent 7753f79 commit ba6f469

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

API.md

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ Tensor Tensor_reciprocal(Tensor self);
346346

347347
These operations reduce a tensor to a single value or along a specified dimension. They are exposed via macros for a simpler API.
348348

349-
### Sum
349+
### `Tensor_sum`
350350

351351
**Usage:**
352352

@@ -360,7 +360,7 @@ Tensor sum_dim = Tensor_sum(my_tensor, 1);
360360

361361
#### **Underlying Functions:** `Tensor_sum_all(Tensor self)`, `Tensor_sum_dim(Tensor self, int dim)`
362362

363-
### Mean
363+
### `Tensor_mean`
364364

365365
**Usage:**
366366

@@ -374,7 +374,7 @@ Tensor mean_dim = Tensor_mean(my_tensor, 1);
374374

375375
#### **Underlying Functions:** `Tensor_mean_all(Tensor self)`, `Tensor_mean_dim(Tensor self, int dim)`
376376

377-
### Max
377+
### `Tensor_max`
378378

379379
**Usage:**
380380

@@ -390,7 +390,7 @@ Tensor max_indices = max_res.indices;
390390

391391
#### **Underlying Functions:** `Tensor_max_all(Tensor self)`, `TensorMaxMinResult Tensor_max_dim(Tensor self, int dim)`
392392

393-
### Min
393+
### `Tensor_min`
394394

395395
**Usage:**
396396

@@ -404,8 +404,6 @@ TensorMaxMinResult min_res = Tensor_min(my_tensor, 1);
404404

405405
#### **Underlying Functions:** `Tensor_min_all(Tensor self)`, `TensorMaxMinResult Tensor_min_dim(Tensor self, int dim)`
406406

407-
### Argmax
408-
409407
### `Tensor_argmax`
410408

411409
Finds the indices of the maximum values along the last dimension.

0 commit comments

Comments
 (0)