Index Expressions
Index expressions are used to express computations in taco. They are formed by indexing into pytaco.tensor
s with
index_var
s and using the variety of arithmetic operations PyTaco provides for index vars.
It should be noted that index expressions describe to taco how a computation should be performed. Thus, they must be assigned to a tensor before taco does any computations.
The documentation in this section displays all the functions taco supports with index expressions as well as a wide number of examples for using them.
Index expressions provide a convenient way for taco to construct outputs without making temporaries and are essential to getting leveraging taco’s ability to fuse operations.
There is an equivalent tensor function for each index expression function. The user should note that the tensor functions all use these under the hood.
- PyTaco Index Variables
- Index Expression Objects
- Index Expression Functions
- pytaco.cast
- pytaco.remainder
- pytaco.abs
- pytaco.pow
- pytaco.square
- pytaco.cube
- pytaco.sqrt
- pytaco.cube_root
- pytaco.exp
- pytaco.log
- pytaco.log10
- pytaco.sin
- pytaco.cos
- pytaco.tan
- pytaco.asin
- pytaco.acos
- pytaco.atan
- pytaco.atan2
- pytaco.sinh
- pytaco.cosh
- pytaco.tanh
- pytaco.asinh
- pytaco.acosh
- pytaco.atanh
- pytaco.logical_not
- pytaco.gt
- pytaco.lt
- pytaco.ge
- pytaco.le
- pytaco.eq
- pytaco.ne
- pytaco.max
- pytaco.min
- pytaco.heaviside
- pytaco.sum
- pytaco.add
- pytaco.mul
- pytaco.sub
- pytaco.div
- Convenience Functions