search

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
https://cg.cs.tsinghua.edu.cn/jittor/
Apache License 2.0
3.07k stars 307 forks source link

fix: jt.Var.expand with valid index -1 #546

Closed Hanyx2021 closed 3 months ago

Hanyx2021 commented 3 months ago

fix jt.Var.expand function which has valid parameter -1 example:

`import jittor as jt

a = jt.rand(3,3) b = a.expand(1,3,-1) print(b.shape)`