Welcome to dtorch’s documentation!
Description
DTorch is a package made by a student at Epitech to improve his understanding of pytorch and better his knowledge of ia.
It is built on top of numpy which is a scientific framework for math between matrices.
This project run on cpu but still have decent computation time making it usable for model building, optimization and saving/loading. The tensors created can work with numpy but remember that the gradients will not be calculated if the operation are not in the range of control of the library. Therefore, a usage of the packages methods on the tensors if preferable. Cuda support may appear in the future and similarly for the mkldnn library that seems to be excellent when working on CPU.
Tip
A direct advantage of using dtorch is it’s lightness. The package is currently close to 14 Ko and is fast to load in any project while the use of torch often lead to a slow start.
Content
- dtorch
- Description
- Modules
- dtorch.jtensors
JTensors
JTensors.grad
JTensors.require_grads
JTensors.ndims
JTensors.shape
JTensors.dtype
JTensors.T
JTensors.itemsize
JTensors.size
JTensors.stride
JTensors.backward()
JTensors.numpy()
JTensors.transpose()
JTensors.reshape()
JTensors.max()
JTensors.sum()
JTensors.rearrange()
JTensors.shuffle()
JTensors.norm()
JTensors.detach()
JTensors.clone()
JTensors.unsqueeze()
JTensors.squeeze()
- dtorch.nn
- dtorch.functionnal
- dtorch.einops
- dtorch.optim
- dtorch.loss
- dtorch.jtensors
- dtorchtree
- dtorchvision
- dtorchtext