tensor.narrow()函数

最新推荐文章于 2025-05-08 19:54:43 发布

Shashank497

最新推荐文章于 2025-05-08 19:54:43 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签：深度学习 pytorch 机器学习

本文链接：https://blog.csdn.net/EMIvv/article/details/122590890

pytorch 专栏收录该内容

6 篇文章

订阅专栏

本文详细介绍了PyTorch中narrow函数的使用方法，该函数用于选取张量特定维度上的子区间。通过具体实例展示了如何指定维度、起始位置及长度来获取所需的数据片段。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

input.narrow(dimension, start, length) → Tensor
  # 表示取变量input在第dimension维上，从索引start到start+length范围（不包括start+length）的值。
example:
In [2]: x = torch.Tensor([[1,2,3], [4,5,6], [7,8,9]])
In [4]: x.narrow(0,0,3)
Out[4]: 
tensor([[1., 2., 3.],
        [4., 5., 6.],
        [7., 8., 9.]])
In [6]: x.narrow(0,1,2)
Out[6]: 
tensor([[4., 5., 6.],
        [7., 8., 9.]])
 
In [7]: x.narrow(1, 1, 2)
Out[7]: 
tensor([[2., 3.],
        [5., 6.],
        [8., 9.]])