Skip to content

Commit 57617ee

Browse files
pcpLiufacebook-github-bot
authored andcommitted
Replace resize_dim() with set_sizes_and_strides() (pytorch#17127)
Summary: Pull Request resolved: pytorch#17127 Replace resize_dim() with set_sizes_and_strides() in `THTensor_(squeeze1d) in aten/src/TH/generic/THTensor.cpp` Reviewed By: ezyang Differential Revision: D14088697 fbshipit-source-id: 518b72f7c0c4fbedf11a29a6ceb9fee8eefd9273
1 parent 9477c14 commit 57617ee

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

aten/src/TH/generic/THTensor.cpp

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -467,12 +467,20 @@ void THTensor_(squeeze1d)(THTensor *self, THTensor *src, int dimension)
467467

468468
if(src->size(dimension) == 1)
469469
{
470+
std::vector<int64_t> newSize(self->dim() - 1);
471+
std::vector<int64_t> newStride(self->dim() - 1);
472+
for (d = 0; d < dimension; d++)
473+
{
474+
newSize[d] = self->size(d);
475+
newStride[d] = self->stride(d);
476+
}
477+
470478
for(d = dimension; d < self->dim()-1; d++)
471479
{
472-
self->set_size(d, self->size(d+1));
473-
self->set_stride(d, self->stride(d+1));
480+
newSize[d] = self->size(d+1);
481+
newStride[d] = self->stride(d+1);
474482
}
475-
self->resize_dim((unsigned int)(self->dim() - 1));
483+
self->set_sizes_and_strides(newSize, newStride);
476484
}
477485
}
478486

0 commit comments

Comments
 (0)