torch_npu.npu_stride_add

torch_npu.npu_stride_add(x1, x2, offset1, offset2, c1_len) -> Tensor

添加两个张量的partial values，格式为NC1HWC0。

• 参数解释：
  • x1 (Tensor) - 5HD张量。
  • x2 (Tensor) - 与“x1”类型相同shape相同（C1值除外）的张量。
  • offset1 (Scalar) - “x1”中C1的offset value。
  • offset2 (Scalar) - “x2”中C1的offset value。
  • c1_len (Scalar) - “y”的C1 len。该值必须小于“x1”和“x2”中C1与offset的差值。
• 约束条件：

  无

• 示例：

  >>> a=torch.tensor([[[[[1.]]]]]).npu()
  >>> b=torch_npu.npu_stride_add(a, a, 0, 0, 1)
  >>> btensor([[[[[2.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]],
          [[[0.]]]]], device='npu:0')