Files
tinygrad/docs/tensor/movement.md
geohotstan 72a41095bc add Tensor.meshgrid (#7714)
* initial implementation and test

* some other places that can use meshgrid

* revert the onnx_ops change

* add to docs

* revert interpolate too

* update

* improve edge case test

* might as well test grad

* add to test can improve docs

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
2024-11-16 23:06:47 -05:00

627 B

Movement (low level)

::: tinygrad.Tensor.view ::: tinygrad.Tensor.reshape ::: tinygrad.Tensor.expand ::: tinygrad.Tensor.permute ::: tinygrad.Tensor.flip ::: tinygrad.Tensor.shrink ::: tinygrad.Tensor.pad

Movement (high level)

::: tinygrad.Tensor.gather ::: tinygrad.Tensor.cat ::: tinygrad.Tensor.stack ::: tinygrad.Tensor.repeat ::: tinygrad.Tensor.repeat_interleave ::: tinygrad.Tensor.split ::: tinygrad.Tensor.chunk ::: tinygrad.Tensor.meshgrid ::: tinygrad.Tensor.squeeze ::: tinygrad.Tensor.unsqueeze ::: tinygrad.Tensor.T ::: tinygrad.Tensor.transpose ::: tinygrad.Tensor.flatten ::: tinygrad.Tensor.unflatten