27#include "neml2/user_tensors/UserTensorBase.h"
const TensorShape _batch_sizes
Batch shape.
Definition FactoryMethodBase.h:46
FactoryMethodBase(const OptionSet &options)
const TensorShape _base_sizes
Base shape.
Definition FactoryMethodBase.h:49
const unsigned int _intmd_dim
Intermediate dimension.
Definition FactoryMethodBase.h:52
static OptionSet expected_options()
A custom map-like data structure. The keys are strings, and the values can be nonhomogeneously typed.
Definition OptionSet.h:52
UserTensorBase(const OptionSet &options)
Definition DiagnosticsInterface.h:31
c10::SmallVector< Size, 8 > TensorShape
Definition types.h:72