semvec/pypi-packages huggingface dataset:
pamit 0.1.2
Taichi Map Utils
Requires Python: >=3.6

# pamit
![Tests](https://github.com/ifsheldon/pamit/actions/workflows/run_tests.yaml/badge.svg)

pamit -> ti map: Taichi Map Utils

**PRs are welcomed! Please see TODOs.**

## Installation and Dependency
To install `pamit`, enter

`python -m pip install pamit`

Make sure you have `Taichi` installed.

## Usage

The APIs of `map_inplace()` and `map_to()` are straight forward.

`map_to` requires the target field to have the same shape of the source field, but the "channels" may differ. Please see `examples/extend_dimension` for example.

If `with_index` is `True`, then the Taichi function is expected to have `index` (i.e., the index of an element in a field) and `element` (i.e., the value of an element in a field) as first two arguments.

Otherwise, the Taichi function is expected to have `element` as the first argument.

```python
from pamit import map_inplace
import taichi as ti


@ti.func
def increment(element, val):
    return element + val


field = ti.field(ti.f32, shape=(2, 3))
map_inplace(field, increment, False, 2.0)
print(field)
```

## TODOs
* Check backward capability
* Reduce function
* Simple user documentation