| #pragma once | |
| #include <c10/core/TensorOptions.h> | |
| namespace at { | |
| // Represents the initial TensorOptions, before the "defaults" are ever changed. | |
| // This is designed to be used in library code, where the explicit devices, | |
| // dtypes, etc. are known. NOTE: this is not a stable API. | |
| inline TensorOptions initialTensorOptions() { | |
| return TensorOptions(kCPU).dtype(kFloat).layout(kStrided).requires_grad( | |
| false); | |
| } | |
| } // namespace at |