| 1234567891011121314151617181920 |
- #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
- #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
- #else
- #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
- #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
