Pytorch——narrow()切片函数详解

narrow():用于tensor切片

一、举例

weight = bg_score.narrow(1, 1, 1) 

二、解释

  • 对tensor类型的变量bg_score进行切片
  • 第一个1:对列操作(0为对行操作)
  • 第二个1:从第2列开始
  • 第三个1:取1列

参考链接

  1. pytorch 中narrow的使用
  2. torch.narrow()函数使用详解

你可能感兴趣的:(PyTorch,pytorch,narrow)