torch_npu.npu_pad

API接口

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)