torch_npu.npu_slice

torch_npu.npu_slice(self, offsets, size) -> Tensor

从张量中提取切片。

• 参数解释：
  • self (Tensor) - 输入张量。
  • offsets (ListInt) - 数据类型：int32，int64。
  • size (ListInt) - 数据类型：int32，int64。
• 约束条件：

  无

• 示例：

  >>> input = torch.tensor([[1,2,3,4,5], [6,7,8,9,10]], dtype=torch.float16).to("npu")
  >>> offsets = [0, 0]>>> size = [2, 2]
  >>> output = torch_npu.npu_slice(input, offsets, size)
  >>> output
  tensor([[1., 2.],
          [6., 7.]], device='npu:0', dtype=torch.float16)