[functorch] replace reshape with view (pytorch/functorch#304)

kshitij12345 · zou3519 · commit c56f769062c9 · 2022-07-20T15:43:29.000-07:00
diff --git a/functorch/functorch/csrc/BatchRulesScatterOps.cpp b/functorch/functorch/csrc/BatchRulesScatterOps.cpp
@@ -387,7 +387,7 @@ Tensor get_expanded_index(const Tensor& index, IntArrayRef self_size, int64_t di
   {
     VmapDimVector new_index_shape(self_size.size(), 1);
     new_index_shape[dim] = idx_size;
-    index_ = index.reshape(new_index_shape);
+    index_ = index.view(new_index_shape);
   }
   // Now apply expand to index_
   {

Original file line number	Diff line number	Diff line change
`@@ -387,7 +387,7 @@ Tensor get_expanded_index(const Tensor& index, IntArrayRef self_size, int64_t di`
`387`	`387`	`{`
`388`	`388`	`VmapDimVector new_index_shape(self_size.size(), 1);`
`389`	`389`	`new_index_shape[dim] = idx_size;`
`390`		`- index_ = index.reshape(new_index_shape);`
	`390`	`+ index_ = index.view(new_index_shape);`
`391`	`391`	`}`
`392`	`392`	`// Now apply expand to index_`
`393`	`393`	`{`