torch_npu.npu_pad

torch_npu.npu_pad(input, paddings) -> Tensor

填充张量。

• 参数解释：
  • input (Tensor) - 输入张量。
  • paddings (ListInt) - 数据类型：int32、int64。
• 约束条件：

  无

• 示例：

  >>> input = torch.tensor([[20, 20, 10, 10]], dtype=torch.float16).to("npu")
  >>> paddings = [1, 1, 1, 1]
  >>> output = torch_npu.npu_pad(input, paddings)
  >>> output
  tensor([[ 0.,  0.,  0.,  0.,  0.,  0.],
          [ 0., 20., 20., 10., 10.,  0.],
          [ 0.,  0.,  0.,  0.,  0.,  0.]], device='npu:0', dtype=torch.float16)