lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

updated docs

This commit is contained in:
lzhengning 2021-09-30 16:04:29 +08:00
parent 5c2fb6bfce
commit e57f0e4879
7 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.cn.md
View File

@ -1,5 +1,6 @@
# Jittor: 即时编译深度学习框架
![Jittor Logo](doc/logo.png)
[快速开始](#快速开始) | [安装](#安装) | [教程](#教程)

2
README.md
View File

@ -1,5 +1,7 @@
# Jittor: a Just-in-time(JIT) deep learning framework
![Jittor Logo](doc/logo.png)
[Quickstart](#quickstart) | [Install](#install) | [Tutorial](#tutorial) | [Chinese](./README.cn.md)

2
README.src.md
View File

@ -1,6 +1,8 @@
# Jittor: a Just-in-time(JIT) deep learning framework
# Jittor: 即时编译深度学习框架
![Jittor Logo](doc/logo.png)
[Quickstart](#quickstart) | [Install](#install) | [Tutorial](#tutorial) | [Chinese](./README.cn.md)
[快速开始](#快速开始) | [安装](#安装) | [教程](#教程)

BIN
doc/logo.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 28 KiB

6
python/jittor/misc.py
View File

@ -214,7 +214,8 @@ def t(x):
return x.transpose(*pose)
jt.Var.t = t
def median(x,dim=None,keepdim=False):
def median(x,dim=None,keepdim=False, keepdims=False):
keepdim = keepdim or keepdims
if dim is None:
x = x.reshape(-1)
dim=0
@ -637,7 +638,8 @@ def topk(input, k, dim=None, largest=True, sorted=True):
jt.Var.topk = topk
def kthvalue(input, k, dim=None, keepdim=False):
def kthvalue(input, k, dim=None, keepdim=False, keepdims=False):
keepdim = keepdim or keepdims
if dim is None:
dim = -1
if dim<0:

14
python/jittor/src/ops/reduce_op.cc
View File

@ -36,7 +36,7 @@ unordered_set<string> reduce_ops = {
* [in] dim: int or tuples of ints (optional). If specified, reduce along the given the dimension(s).
* [in] keepdim: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
* [in] keepdims: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
----------------
@ -67,7 +67,7 @@ unordered_set<string> reduce_ops = {
* [in] dim: int or tuples of ints (optional). If specified, reduce along the given the dimension(s).
* [in] keepdim: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
* [in] keepdims: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
----------------
@ -98,7 +98,7 @@ unordered_set<string> reduce_ops = {
* [in] dim: int or tuples of ints (optional). If specified, reduce along the given the dimension(s).
* [in] keepdim: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
* [in] keepdims: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
----------------
@ -129,7 +129,7 @@ unordered_set<string> reduce_ops = {
* [in] dim: int or tuples of ints (optional). If specified, reduce along the given the dimension(s).
* [in] keepdim: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
* [in] keepdims: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
----------------
@ -160,7 +160,7 @@ unordered_set<string> reduce_ops = {
* [in] dim: int or tuples of ints (optional). If specified, reduce along the given the dimension(s).
* [in] keepdim: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
* [in] keepdims: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
----------------
@ -191,7 +191,7 @@ unordered_set<string> reduce_ops = {
* [in] dim: int or tuples of ints (optional). If specified, reduce along the given the dimension(s).
* [in] keepdim: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
* [in] keepdims: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
----------------
@ -226,7 +226,7 @@ unordered_set<string> reduce_ops = {
* [in] dim: int or tuples of ints (optional). If specified, reduce along the given the dimension(s).
* [in] keepdim: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
* [in] keepdims: bool (optional). Whether the output has ``dim`` retained or not. Defaults to be False.
----------------

2
python/jittor/src/ops/reindex_reduce_op.h
View File

@ -56,7 +56,7 @@ struct ReindexReduceOp : Op {
* [in] shape: the output shape, a integer array
* [in] indexes: array of c++ style integer expression, its length should be the same with length of shape, some buildin variables it can use are::
* [in] indexes: array of c++ style integer expression, its length should be the same with length of output shape, some buildin variables it can use are::
XDIM, xshape0, ..., xshapem, xstride0, ..., xstridem
YDIM, yshape0, ..., yshapen, ystride0, ..., ystriden