title: Pytorch select tensor
body: <p>I want to know if Pytorch have a slice function (same as tf). In particular, I want to select the orange color rows. </p>
  
  <p><a href="https://i.stack.imgur.com/Z7kKQ.jpg" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/Z7kKQ.jpg" alt="enter image description here"></a></p>