| namespace at { namespace native { | |
| // This function makes tensors which have zero stride contiguous, by | |
| // setting the strides to 1. | |
| inline Tensor contiguousIfZeroInStrides(const Tensor& t) { | |
| for (auto s : t.strides()) { | |
| if (s == 0) return t.contiguous(); | |
| } | |
| return t; | |
| } | |
| }} | |