2025-03-08 09:09:31 +00:00
|
|
|
## Reduce
|
|
|
|
|
|
|
|
::: tinygrad.Tensor.sum
|
|
|
|
::: tinygrad.Tensor.prod
|
|
|
|
::: tinygrad.Tensor.max
|
|
|
|
::: tinygrad.Tensor.min
|
|
|
|
::: tinygrad.Tensor.any
|
|
|
|
::: tinygrad.Tensor.all
|
2025-04-18 20:38:55 +09:00
|
|
|
::: tinygrad.Tensor.isclose
|
2025-03-08 09:09:31 +00:00
|
|
|
::: tinygrad.Tensor.mean
|
|
|
|
::: tinygrad.Tensor.var
|
2025-04-18 20:38:55 +09:00
|
|
|
::: tinygrad.Tensor.var_mean
|
2025-03-08 09:09:31 +00:00
|
|
|
::: tinygrad.Tensor.std
|
|
|
|
::: tinygrad.Tensor.std_mean
|
|
|
|
::: tinygrad.Tensor.softmax
|
|
|
|
::: tinygrad.Tensor.log_softmax
|
|
|
|
::: tinygrad.Tensor.logsumexp
|
|
|
|
::: tinygrad.Tensor.logcumsumexp
|
|
|
|
::: tinygrad.Tensor.argmax
|
|
|
|
::: tinygrad.Tensor.argmin
|
|
|
|
|
|
|
|
## Processing
|
|
|
|
|
|
|
|
::: tinygrad.Tensor.avg_pool2d
|
|
|
|
::: tinygrad.Tensor.max_pool2d
|
2025-04-18 20:38:55 +09:00
|
|
|
::: tinygrad.Tensor.max_unpool2d
|
2025-03-08 09:09:31 +00:00
|
|
|
::: tinygrad.Tensor.conv2d
|
|
|
|
::: tinygrad.Tensor.conv_transpose2d
|
|
|
|
::: tinygrad.Tensor.dot
|
|
|
|
::: tinygrad.Tensor.matmul
|
|
|
|
::: tinygrad.Tensor.einsum
|
|
|
|
::: tinygrad.Tensor.cumsum
|
|
|
|
::: tinygrad.Tensor.cummax
|
|
|
|
::: tinygrad.Tensor.triu
|
|
|
|
::: tinygrad.Tensor.tril
|
|
|
|
::: tinygrad.Tensor.interpolate
|
|
|
|
::: tinygrad.Tensor.scatter
|
2025-04-18 20:38:55 +09:00
|
|
|
::: tinygrad.Tensor.scatter_reduce
|
|
|
|
::: tinygrad.Tensor.masked_select
|
2025-06-22 10:51:42 +09:00
|
|
|
::: tinygrad.Tensor.masked_fill
|
2025-04-18 20:38:55 +09:00
|
|
|
::: tinygrad.Tensor.sort
|
|
|
|
::: tinygrad.Tensor.topk
|
2025-06-22 10:51:42 +09:00
|
|
|
::: tinygrad.Tensor.multinomial
|
2025-03-08 09:09:31 +00:00
|
|
|
|
|
|
|
## Neural Network (functional)
|
|
|
|
|
|
|
|
::: tinygrad.Tensor.linear
|
|
|
|
::: tinygrad.Tensor.sequential
|
|
|
|
::: tinygrad.Tensor.layernorm
|
|
|
|
::: tinygrad.Tensor.batchnorm
|
|
|
|
::: tinygrad.Tensor.dropout
|
|
|
|
::: tinygrad.Tensor.one_hot
|
|
|
|
::: tinygrad.Tensor.scaled_dot_product_attention
|
|
|
|
::: tinygrad.Tensor.binary_crossentropy
|
|
|
|
::: tinygrad.Tensor.binary_crossentropy_logits
|
|
|
|
::: tinygrad.Tensor.sparse_categorical_crossentropy
|
|
|
|
::: tinygrad.Tensor.cross_entropy
|
|
|
|
::: tinygrad.Tensor.nll_loss
|