提交 5d755898 编写于 作者: P P2Oileen

fix(funtional): correct the typo: witdth to width

上级 719b5e69
......@@ -1754,7 +1754,7 @@ def sliding_window_transpose(
def pad(
src: Tensor,
pad_witdth: Tuple[Tuple[int, int], ...],
pad_width: Tuple[Tuple[int, int], ...],
mode: str = "constant",
constant_value: float = 0.0,
) -> Tensor:
......@@ -1769,9 +1769,9 @@ def pad(
if mode.lower() == "edge":
mode = "replicate"
for i in range(0, len(pad_witdth)):
p_offsets[i * 2] = pad_witdth[i][0]
p_offsets[i * 2 + 1] = pad_witdth[i][1]
for i in range(0, len(pad_width)):
p_offsets[i * 2] = pad_width[i][0]
p_offsets[i * 2 + 1] = pad_width[i][1]
op = builtin.Padding(
front_offset_dim0=p_offsets[0],
......
......@@ -12,16 +12,16 @@ class Pad(Module):
def __init__(
self,
pad_witdth: Tuple[Tuple[int, int], ...],
pad_width: Tuple[Tuple[int, int], ...],
mode: str = "constant",
constant_val: float = 0.0,
):
super().__init__()
self.pad_width = pad_witdth
self.pad_width = pad_width
self.mode = mode
self.pad_val = constant_val
def forward(self, src):
return nn.pad(
src, pad_witdth=self.pad_width, mode=self.mode, constant_value=self.pad_val
src, pad_width=self.pad_width, mode=self.mode, constant_value=self.pad_val
)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册