869b558268f198de90822e4ea17182dc74b78213,sonnet/python/modules/conv.py,Conv2D,_build,#Conv2D#Any#,315

Before Change

        if mask_shape != self._kernel_shape:
          raise base.IncompatibleShapeError(
              "Invalid mask shape: {}".format(mask_shape))
        mask = np.reshape(self._mask, self._kernel_shape + (1, 1))
      elif mask_rank == 4:
        if mask_shape != tuple(weight_shape):
          raise base.IncompatibleShapeError(
              "Invalid mask shape: {}".format(mask_shape))

After Change

          raise base.IncompatibleShapeError(
              "Invalid mask shape: {}".format(tuple(mask_shape)))
        mask = tf.expand_dims(tf.expand_dims(mask, -1), -1)
      elif mask_shape != list(weight_shape):
        raise base.IncompatibleShapeError(
            "Invalid mask shape: {}".format(tuple(mask_shape)))
      w *= mask

    outputs = tf.nn.convolution(inputs, w, strides=self._stride,

Instances