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

API documentation bug in `jt.nn.conv_transpose2d` #532

Open x0w3n opened 3 months ago

x0w3n commented 3 months ago

Describe the bug

The documentation description for jt.nn.conv_transpose2d has the function nn.conv_transpose instead of conv_transpose2d in the Code Examples section, which may cause user inconvenience. https://cg.cs.tsinghua.edu.cn/jittor/assets/docs/jittor.nn.html#jittor.nn.conv_transpose2d

Full Log

import jittor as jt
from jittor import nn
x = jt.ones(2,3,4,5)            
w = jt.ones(3,6,3,3)             
nn.conv_transpose(x,w).shape

Minimal Reproduce

Expected behavior

Should use nn.conv_transpose2 instead of nn.conv_transpose