1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| #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).is_variable(false);
| }
|
| }
|
|
