pytorch index_select用法

        在看动手学深度学习 pytorch版本的时候,看到其中使用了index_select方法。感觉这个方法较为常用和有用。所以需要弄懂。 用法如下:         torch.index_select(input, dim, index, out=None) → Tensor 参数input(Tensor):表示被选择的tensor 参数dim(int):表示是在哪个维度做选择。为1
相关文章
相关标签/搜索