186d078f6f9de9d7f50da14b29166b77290b93ca,nussl/ml/networks/modules.py,Embedding,forward,#Embedding#,229

Before Change

        shape = data.shape
        data = data.transpose(self.dim_to_embed, -1)
        data = self.linear(data)
        data = data.transpose(-1, self.dim_to_embed)

        shape = shape[:-1] + (
            self.num_features, self.num_audio_channels, self.embedding_size,)
        data = data.reshape(shape)

After Change

            data = data.transpose(_dim, -1)

        // the new shape doesn"t have the embedded dimensions
        shape = [
            v for i, v in enumerate(shape) 
            if i not in self.dim_to_embed
        ]
        
        shape = tuple(shape)
        data = data.reshape(shape + (-1,))
        data = self.linear(data)

        shape = shape + (

Instances