提交 62bee2ff 编写于 作者: S ShusenTang
...@@ -164,7 +164,7 @@ tensor([1.6035, 1.8110, 0.9549]) ...@@ -164,7 +164,7 @@ tensor([1.6035, 1.8110, 0.9549])
|:---:|:---:| |:---:|:---:|
|index_select(input, dim, index)|在指定维度dim上选取,比如选取某些行、某些列| |index_select(input, dim, index)|在指定维度dim上选取,比如选取某些行、某些列|
|masked_select(input, mask)|例子如上,a[a>0],使用ByteTensor进行选取| |masked_select(input, mask)|例子如上,a[a>0],使用ByteTensor进行选取|
|non_zero(input)| 非0元素的下标| |nonzero(input)| 非0元素的下标|
|gather(input, dim, index)|根据index,在dim维度上选取数据,输出的size与index一样| |gather(input, dim, index)|根据index,在dim维度上选取数据,输出的size与index一样|
这里不详细介绍,用到了再查官方文档。 这里不详细介绍,用到了再查官方文档。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册