e29ea9571ae112b5f628a3fc4780a6a8adae45e5,art/defences/thermometer_encoding.py,ThermometerEncoding,__call__,#ThermometerEncoding#Any#Any#,59

Before Change

        :rtype: `np.ndarray`
        
        result = []
        for c in range(x.shape[-1]):
            result.append(self._perchannel(x[:, :, :, c]))

        result = np.concatenate(result, axis=3)
        result = np.clip(result, self.clip_values[0], self.clip_values[1])

        return result.astype(NUMPY_DTYPE), y

After Change

        result = np.apply_along_axis(self._perchannel, self.channel_index, x)

        // result = np.concatenate(result, axis=self.channel_index)
        if hasattr(self, "clip_values") and self.clip_values is not None:
            np.clip(result, self.clip_values[0], self.clip_values[1], out=result)

        return result.astype(NUMPY_DTYPE), y

    def _perchannel(self, x):
        

Instances