Coordinate offsets

  • A Taichi tensor can be defined with coordinate offsets. The offsets will move tensor bounds so that tensor origins are no longer zero vectors. A typical use case is to support voxels with negative coordinates in physical simulations.
  • For example, a matrix of 32x64 elements with coordinate offset (-16, 8) can be defined as the following:
a = ti.Matrix(2, 2, dt=ti.f32, shape=(32, 64), offset=(-16, 8))

In this way, the tensor’s indices are from (-16, 8) to (16, 72) (exclusive).

a[-16, 32]  # lower left corner
a[16, 32]   # lower right corner
a[-16, 64]  # upper left corner
a[16, 64]   # upper right corner

注解

The dimensionality of tensor shapes should be consistent with that of the offset. Otherwise, a AssertionError will be raised.

a = ti.Matrix(2, 3, dt=ti.f32, shape=(32,), offset=(-16, ))          # Works!
b = ti.Vector(3, dt=ti.f32, shape=(16, 32, 64), offset=(7, 3, -4))   # Works!
c = ti.Matrix(2, 1, dt=ti.f32, shape=None, offset=(32,))             # AssertionError
d = ti.Matrix(3, 2, dt=ti.f32, shape=(32, 32), offset=(-16, ))       # AssertionError
e = ti.var(dt=ti.i32, shape=16, offset=-16)                          # Works!
f = ti.var(dt=ti.i32, shape=None, offset=-16)                        # AssertionError
g = ti.var(dt=ti.i32, shape=(16, 32), offset=-16)                    # AssertionError