geohotstan
585f3a0f24
Add isinf and isnan ops to Tensor ( #7484 )
...
* move isinf and isnan to new branch
* sneak a roll documentation fix in
* add to docs
* update test coverage for detect_positive and detect_negative
* add types to isinf args
2024-11-02 12:12:52 -04:00
geohotstan
6513690223
Add Tensor.hardsigmoid ( #7433 )
...
* move hardsigmoid to new branch
* add to test
* add NOTE to mention differing values for alpha and beta that match torch
* shift from relu6
* correct shift implementation
* or we just use relu? no more 666
2024-11-01 08:36:52 -04:00
chenyu
fb694a63eb
Tensor.erf ( #7419 )
...
the same one used in onnx and the one in bert.
2024-10-30 18:12:28 -04:00
George Hotz
4438d6a467
Tensor.from_url API [pr] ( #7210 )
...
* Tensor.fetch API [pr]
* update docs
* from_url
2024-10-22 14:54:17 +08:00
jeffzh4ng
19a7e41113
implement logcumsumexp ( #6921 )
...
* implement logcumsumexp
* change axis=None to axis=0
2024-10-06 10:45:36 -04:00
nimlgen
3c56aeee70
add Tensor.from_blob ( #6765 )
...
* draft tensor from pointer init
* some docs and types
* comment
* cleaner
* test
* malloc
* qcom cl interop
* jit example
* cleaner
* dealoc
* wording
* docs
2024-09-26 18:33:19 +08:00
chenyu
590c0922b6
Tensor.prod ( #6250 )
...
* Tensor.prod
a new reduce op!
* onnx ReduceProd
2024-08-23 10:06:32 -04:00
Alessandro Benetti
9328248610
support for std_mean and cross_entropy ( #6181 )
...
* support for std_mean and cross_entropy (#3 )
* Cross entropy and std mean support
* remove extra examples
2024-08-19 12:06:44 -07:00
George Hotz
97c3563109
hotfix: clamp in docs
2024-08-13 16:06:30 -07:00
George Hotz
0a8668cf30
improvements to docs
2024-08-07 09:57:24 -07:00
Eitan Turok
39c8c9c00a
Add docs ( #5942 )
...
* init commit
* finish writing
* add to docs
* fix docs
* fix typo
* delete new line
* rename to tensor properties
---------
Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com >
2024-08-07 07:38:51 -07:00
chenyu
0afcbfae84
docs: add Tensor.interpolate to doc page ( #5510 )
2024-07-16 14:17:19 -04:00
chenyu
6856f915d6
Tensor.any and Tensor.all ( #5320 )
...
does not work in ptx yet due to how boolean tensor is handled
2024-07-07 14:36:00 -04:00
chenyu
c1e330f302
Tensor.int and Tensor.bool ( #5317 )
2024-07-07 11:52:58 -04:00
George Hotz
146eb3a811
hotfix: add repeat_interleave docs
2024-06-30 15:25:18 -07:00
chenyu
20b50d8d64
doc: manual_seed ( #4987 )
...
there was a docstring just not linked to the doc page. also updated the example to show re-seed instead of a internal variable
2024-06-15 19:57:26 -04:00
chenyu
e22cdb40f3
docs: fix mkdoc warnings and link to tensor.md ( #4760 )
2024-05-28 14:24:11 -04:00
wozeparrot
b2b49cef6f
split tensor docs ( #4754 )
2024-05-28 11:03:52 -07:00
