27#include "neml2/base/config.h"
29#include <torch/types.h>
39typedef std::vector<at::indexing::TensorIndex>
TorchSlice;
The wrapper (decorator) for cross-referencing unresolved values at parse time.
Definition CrossRef.h:52
Definition CrossRef.cxx:32
const torch::TensorOptions default_tensor_options()
Definition types.cxx:30
int64_t TorchSize
Definition types.h:35
std::vector< TorchSize > TorchShape
Definition types.h:36
double Real
Definition types.h:33
torch::IntArrayRef TorchShapeRef
Definition types.h:37
at::indexing::TensorIndex TorchIndex
Definition types.h:38
const torch::TensorOptions default_integer_tensor_options()
We similarly want to have a default integer scalar type for some types of tensors.
Definition types.cxx:42
std::vector< at::indexing::TensorIndex > TorchSlice
Definition types.h:39
int Integer
Definition types.h:34