提交 62bee2ff 编写于 作者: S ShusenTang
......@@ -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一样|
这里不详细介绍,用到了再查官方文档。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册