Skip to content

Commit 62bee2f

Browse files
committed
2 parents e981295 + 004905e commit 62bee2f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/chapter02_prerequisite/2.2_tensor.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ tensor([1.6035, 1.8110, 0.9549])
164164
|:---:|:---:|
165165
|index_select(input, dim, index)|在指定维度dim上选取,比如选取某些行、某些列|
166166
|masked_select(input, mask)|例子如上,a[a>0],使用ByteTensor进行选取|
167-
|non_zero(input)| 非0元素的下标|
167+
|nonzero(input)| 非0元素的下标|
168168
|gather(input, dim, index)|根据index,在dim维度上选取数据,输出的size与index一样|
169169

170170
这里不详细介绍,用到了再查官方文档。

0 commit comments

Comments
 (0)