Skip to content

Commit

Permalink
docs: move down tinygrad op and add missing methods (#4710)
Browse files Browse the repository at this point in the history
  • Loading branch information
chenyuxyz committed May 24, 2024
1 parent 12ec02d commit a921f33
Showing 1 changed file with 19 additions and 17 deletions.
36 changes: 19 additions & 17 deletions docs/tensor.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,22 +10,6 @@
::: tinygrad.Tensor.dtype
::: tinygrad.Tensor.device

## tinygrad ops

::: tinygrad.Tensor.schedule_with_vars
::: tinygrad.Tensor.schedule
::: tinygrad.Tensor.realize
::: tinygrad.Tensor.replace
::: tinygrad.Tensor.assign
::: tinygrad.Tensor.detach
::: tinygrad.Tensor.to
::: tinygrad.Tensor.to_
::: tinygrad.Tensor.shard
::: tinygrad.Tensor.shard_
::: tinygrad.Tensor.contiguous
::: tinygrad.Tensor.contiguous_backward
::: tinygrad.Tensor.backward

## Creation (basic)

::: tinygrad.Tensor.empty
Expand All @@ -42,6 +26,7 @@

::: tinygrad.Tensor.rand
::: tinygrad.Tensor.randn
::: tinygrad.Tensor.randint
::: tinygrad.Tensor.normal
::: tinygrad.Tensor.uniform
::: tinygrad.Tensor.scaled_uniform
Expand All @@ -56,8 +41,25 @@
::: tinygrad.Tensor.tolist
::: tinygrad.Tensor.numpy

## tinygrad ops

::: tinygrad.Tensor.schedule_with_vars
::: tinygrad.Tensor.schedule
::: tinygrad.Tensor.realize
::: tinygrad.Tensor.replace
::: tinygrad.Tensor.assign
::: tinygrad.Tensor.detach
::: tinygrad.Tensor.to
::: tinygrad.Tensor.to_
::: tinygrad.Tensor.shard
::: tinygrad.Tensor.shard_
::: tinygrad.Tensor.contiguous
::: tinygrad.Tensor.contiguous_backward
::: tinygrad.Tensor.backward

## Movement (low level)

::: tinygrad.Tensor.view
::: tinygrad.Tensor.reshape
::: tinygrad.Tensor.expand
::: tinygrad.Tensor.permute
Expand All @@ -76,6 +78,7 @@
::: tinygrad.Tensor.squeeze
::: tinygrad.Tensor.unsqueeze
::: tinygrad.Tensor.pad2d
::: tinygrad.Tensor.T
::: tinygrad.Tensor.transpose
::: tinygrad.Tensor.flatten
::: tinygrad.Tensor.unflatten
Expand Down Expand Up @@ -181,4 +184,3 @@
::: tinygrad.Tensor.bitcast
::: tinygrad.Tensor.float
::: tinygrad.Tensor.half

0 comments on commit a921f33

Please sign in to comment.