diff --git a/advanced_source/dynamic_quantization_tutorial.py b/advanced_source/dynamic_quantization_tutorial.py index cabdd90d224..cfdacaca329 100644 --- 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): diff --git a/beginner_source/transformer_tutorial.py b/beginner_source/transformer_tutorial.py index aab564391e5..70bc9ccdb29 100644 --- a/beginner_source/transformer_tutorial.py +++ b/beginner_source/transformer_tutorial.py @@ -194,7 +194,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 diff --git a/prototype_source/graph_mode_static_quantization_tutorial.py b/prototype_source/graph_mode_static_quantization_tutorial.py index af649a15f02..64aa81138c6 100644 --- a/prototype_source/graph_mode_static_quantization_tutorial.py +++ b/prototype_source/graph_mode_static_quantization_tutorial.py @@ -137,7 +137,7 @@ def accuracy(output, target, topk=(1,)): res = [] for k in topk: - correct_k = correct[:k].view(-1).float().sum(0, keepdim=True) + correct_k = correct[:k].reshape(-1).float().sum(0, keepdim=True) res.append(correct_k.mul_(100.0 / batch_size)) return res