def construct(self, x):
    x_shape = self.shape(x)
    _check_input_3d(x_shape)
    x = self.expand_dims(x, 2)
    output = self.conv2d(x, self.weight)
    if self.has_bias:
        output = self.bias_add(output, self.bias)
    output = self.squeeze(output)
    return output
def construct(self, x):
    x_shape = self.shape(x)
    _check_input_3d(x_shape)
    x = self.expand_dims(x, 2)
    output = self.conv2d(x, self.weight)
    if self.has_bias:
        output = self.bias_add(output, self.bias)
    output = self.squeeze(output)
    return output