Skip to content

Commit

Permalink
Remainder operator doc (#1836)
Browse files Browse the repository at this point in the history
* Adds remainder ops implementation for Tensor.

* Adds test for % operator.

* Add remainder and % operator entry in tensor.md

---------

Co-authored-by: Jonas Kantic <jk.mail@posteo.net>
  • Loading branch information
Zirconium419122 and kantic authored Jun 1, 2024
1 parent fba1e27 commit e407c76
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions burn-book/src/building-blocks/tensor.md
Original file line number Diff line number Diff line change
Expand Up @@ -228,6 +228,7 @@ Those operations are available for numeric tensor kinds: `Float` and `Int`.
| `tensor.powf_scalar(scalar)` or `tensor.powi_scalar(intscalar)` | `tensor.pow(scalar)` |
| `tensor.prod()` | `tensor.prod()` |
| `tensor.prod_dim(dim)` | `tensor.prod(dim, keepdim=True)` |
| `tensor.rem(other)` or `tensor % other` | `tensor % other` |
| `tensor.scatter(dim, indices, values)` | `tensor.scatter_add(dim, indices, values)` |
| `tensor.select(dim, indices)` | `tensor.index_select(dim, indices)` |
| `tensor.select_assign(dim, indices, values)` | N/A |
Expand Down

0 comments on commit e407c76

Please sign in to comment.