|
|
|
@ -170,7 +170,7 @@ class _BatchNorm(Cell):
|
|
|
|
|
@constexpr
|
|
|
|
|
def _channel_check(channel, num_channel):
|
|
|
|
|
if channel != num_channel:
|
|
|
|
|
raise ValueError("the input channel is not equal with num_channels")
|
|
|
|
|
raise ValueError("the input channel is not equal with num_channel")
|
|
|
|
|
|
|
|
|
|
class BatchNorm1d(_BatchNorm):
|
|
|
|
|
r"""
|
|
|
|
|