diff --git a/docs/chapter02_prerequisite/2.2_tensor.md b/docs/chapter02_prerequisite/2.2_tensor.md index 9294a7740c04393fadd0ee38629803fb06128872..cf35da977075b6831e8687b8c06c5d4c4d652a23 100644 --- a/docs/chapter02_prerequisite/2.2_tensor.md +++ b/docs/chapter02_prerequisite/2.2_tensor.md @@ -164,7 +164,7 @@ tensor([1.6035, 1.8110, 0.9549]) |:---:|:---:| |index_select(input, dim, index)|在指定维度dim上选取,比如选取某些行、某些列| |masked_select(input, mask)|例子如上,a[a>0],使用ByteTensor进行选取| -|non_zero(input)| 非0元素的下标| +|nonzero(input)| 非0元素的下标| |gather(input, dim, index)|根据index,在dim维度上选取数据,输出的size与index一样| 这里不详细介绍,用到了再查官方文档。