Update README.md

README.md

@@ -55,6 +55,8 @@ norm_wav = (raw_wav - mean) / (std+0.000001)
 
 #generate the mask
 mask = torch.ones(1, len(norm_wav))
+
+#batch it (add dim)
 wavs = torch.tensor(norm_wav).unsqueeze(0)