@@ -32,9 +32,9 @@ def constant_padNd(
32
32
33
33
rank = len (input .shape )
34
34
35
- if len (pad ) / 2 > rank :
35
+ if len (pad ) // 2 > rank :
36
36
raise RuntimeError (
37
- f"Trying to pad last { len (pad ) / 2 } dimension but the input only has { rank } dimension."
37
+ f"Trying to pad last { len (pad ) // 2 } dimension but the input only has { rank } dimension."
38
38
)
39
39
40
40
start_list = [0 ] * len (input .shape )
@@ -72,9 +72,9 @@ def reflection_padNd(
72
72
73
73
rank = len (input .shape )
74
74
75
- if len (padding ) / 2 > rank :
75
+ if len (padding ) // 2 > rank :
76
76
raise RuntimeError (
77
- f"Trying to pad last { len (padding ) / 2 } dimension but the input only has { rank } dimension."
77
+ f"Trying to pad last { len (padding ) // 2 } dimension but the input only has { rank } dimension."
78
78
)
79
79
80
80
start_list = [0 ] * len (input .shape )
@@ -110,9 +110,9 @@ def replication_padNd(
110
110
111
111
rank = len (input .shape )
112
112
113
- if len (padding ) / 2 > rank :
113
+ if len (padding ) // 2 > rank :
114
114
raise RuntimeError (
115
- f"Trying to pad last { len (padding ) / 2 } dimension but the input only has { rank } dimension."
115
+ f"Trying to pad last { len (padding ) // 2 } dimension but the input only has { rank } dimension."
116
116
)
117
117
118
118
start_list = [0 ] * len (input .shape )
@@ -148,9 +148,9 @@ def circular_padNd(
148
148
149
149
rank = len (input .shape )
150
150
151
- if len (pad ) / 2 > rank :
151
+ if len (pad ) // 2 > rank :
152
152
raise RuntimeError (
153
- f"Trying to pad last { len (pad ) / 2 } dimension but the input only has { rank } dimension."
153
+ f"Trying to pad last { len (pad ) // 2 } dimension but the input only has { rank } dimension."
154
154
)
155
155
156
156
start_list = [0 ] * len (input .shape )
0 commit comments