123456789101112131415 |
- #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
|