geohotstan cea5853cfa add Tensor.scatter (#7737) ...

* working I think

* where are my onnx scatter tests??

* forward_only for now

* try if nan hack fix NV

* looks like issue is different... CUDA WHY

* oops that was wrong. Try if this fixes CUDA

* simpler multiply

* actually finish this up tmrw morning :x

* fix tests?

* improve tests

* improve test and implementation

* fix ruff

* complete but lots of expected failure...

* reviewed tests

* add onnx tests

* is this a processing op?

* add return type to indicate that it's not in-place

* final cleanups

* use or and improve tests a little

* add masked_index_select

* call it masked_setitem instead

* try

* FIXED

---------

Co-authored-by: chenyu <chenyu@fastmail.com>

2024-11-27 10:52:04 -05:00

..

creation.md

add Tensor.linspace (#7609)

2024-11-12 10:29:36 +08:00

elementwise.md

add inverse trig functions to Tensor (#7805)

2024-11-21 09:13:36 -05:00

index.md

improvements to docs

2024-08-07 09:57:24 -07:00

movement.md

add Tensor.meshgrid (#7714)

2024-11-16 23:06:47 -05:00

ops.md

add Tensor.scatter (#7737)

2024-11-27 10:52:04 -05:00

properties.md

improvements to docs

2024-08-07 09:57:24 -07:00