Skip to content

ychjiang/pytorch

 
 

Repository files navigation

PyTorch Logo


PyTorch is a Python package that provides two high-level features:

  • Tensor computation (like NumPy) with strong GPU acceleration
  • Deep neural networks built on a tape-based autograd system

You can reuse your favorite Python packages such as NumPy, SciPy and Cython to extend PyTorch when needed.

Purpose

中文:该项目的目的是阅读pytorch的后端代码,了解具体的操作实现方式.

English: This project focus on pytorch backword souce code, try to know how is run.

Method

中文:深入pytorch的底层代码,从C/C++角度探索pytorch具体的数据处理操作方式。包括卷积的C++计算代码,调用LAPACK的方式等基本操作。

English:we focus on the pytorch backward source code and try to find the operation of basic data(THTensor),such as Tensor convolution,call LAPACK library.

License

PyTorch is BSD-style licensed, as found in the LICENSE file.

About

These project branch focus on the backward source code of pytorch

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C++ 46.7%
  • Python 30.9%
  • Cuda 14.5%
  • C 4.5%
  • CMake 1.8%
  • Objective-C++ 0.8%
  • Other 0.8%