mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-24 14:28:09 -05:00
f3fdec940d
it's a python style mod. possibily can be cleaner with a floor div relaxed the vmin for MOD slightly for cstyle negatives mod, it's more correct and might fix other bugs
83 lines
2.0 KiB
Markdown
83 lines
2.0 KiB
Markdown
Elementwise ops operate on a per element basis. They don't change the shape of the tensor.
|
|
|
|
## Unary Ops (math)
|
|
|
|
::: tinygrad.Tensor.logical_not
|
|
::: tinygrad.Tensor.neg
|
|
::: tinygrad.Tensor.log
|
|
::: tinygrad.Tensor.log2
|
|
::: tinygrad.Tensor.exp
|
|
::: tinygrad.Tensor.exp2
|
|
::: tinygrad.Tensor.sqrt
|
|
::: tinygrad.Tensor.rsqrt
|
|
::: tinygrad.Tensor.sin
|
|
::: tinygrad.Tensor.cos
|
|
::: tinygrad.Tensor.tan
|
|
::: tinygrad.Tensor.asin
|
|
::: tinygrad.Tensor.acos
|
|
::: tinygrad.Tensor.atan
|
|
::: tinygrad.Tensor.trunc
|
|
::: tinygrad.Tensor.ceil
|
|
::: tinygrad.Tensor.floor
|
|
::: tinygrad.Tensor.round
|
|
::: tinygrad.Tensor.isinf
|
|
::: tinygrad.Tensor.isnan
|
|
::: tinygrad.Tensor.lerp
|
|
::: tinygrad.Tensor.square
|
|
::: tinygrad.Tensor.clamp
|
|
::: tinygrad.Tensor.clip
|
|
::: tinygrad.Tensor.sign
|
|
::: tinygrad.Tensor.abs
|
|
::: tinygrad.Tensor.reciprocal
|
|
|
|
## Unary Ops (activation)
|
|
|
|
::: tinygrad.Tensor.relu
|
|
::: tinygrad.Tensor.sigmoid
|
|
::: tinygrad.Tensor.hardsigmoid
|
|
::: tinygrad.Tensor.elu
|
|
::: tinygrad.Tensor.celu
|
|
::: tinygrad.Tensor.selu
|
|
::: tinygrad.Tensor.swish
|
|
::: tinygrad.Tensor.silu
|
|
::: tinygrad.Tensor.relu6
|
|
::: tinygrad.Tensor.hardswish
|
|
::: tinygrad.Tensor.tanh
|
|
::: tinygrad.Tensor.sinh
|
|
::: tinygrad.Tensor.cosh
|
|
::: tinygrad.Tensor.atanh
|
|
::: tinygrad.Tensor.asinh
|
|
::: tinygrad.Tensor.acosh
|
|
::: tinygrad.Tensor.hardtanh
|
|
::: tinygrad.Tensor.erf
|
|
::: tinygrad.Tensor.gelu
|
|
::: tinygrad.Tensor.quick_gelu
|
|
::: tinygrad.Tensor.leakyrelu
|
|
::: tinygrad.Tensor.mish
|
|
::: tinygrad.Tensor.softplus
|
|
::: tinygrad.Tensor.softsign
|
|
|
|
## Elementwise Ops (broadcasted)
|
|
|
|
::: tinygrad.Tensor.add
|
|
::: tinygrad.Tensor.sub
|
|
::: tinygrad.Tensor.mul
|
|
::: tinygrad.Tensor.div
|
|
::: tinygrad.Tensor.idiv
|
|
::: tinygrad.Tensor.mod
|
|
::: tinygrad.Tensor.xor
|
|
::: tinygrad.Tensor.lshift
|
|
::: tinygrad.Tensor.rshift
|
|
::: tinygrad.Tensor.pow
|
|
::: tinygrad.Tensor.maximum
|
|
::: tinygrad.Tensor.minimum
|
|
::: tinygrad.Tensor.where
|
|
|
|
## Casting Ops
|
|
|
|
::: tinygrad.Tensor.cast
|
|
::: tinygrad.Tensor.bitcast
|
|
::: tinygrad.Tensor.float
|
|
::: tinygrad.Tensor.half
|
|
::: tinygrad.Tensor.int
|
|
::: tinygrad.Tensor.bool |