Update dynamic_quantization_tutorial.py

albanD · web-flow · commit eeff86585bcc · 2020-10-27T13:10:30.000-04:00
diff --git a/advanced_source/dynamic_quantization_tutorial.py b/advanced_source/dynamic_quantization_tutorial.py
@@ -213,7 +213,7 @@ def batchify(data, bsz):
 def get_batch(source, i):
     seq_len = min(bptt, len(source) - 1 - i)
     data = source[i:i+seq_len]
-    target = source[i+1:i+1+seq_len].view(-1)
+    target = source[i+1:i+1+seq_len].reshape(-1)
     return data, target
 
 def repackage_hidden(h):
