27#include "neml2/tensors/Tensor.h"
Definition DiagnosticsInterface.h:31
c10::Device Device
Definition types.h:69
ValueMap valuemap_no_operation(ValueMap &&x)
No operation.
ValueMap valuemap_cat_reduce(const std::vector< ValueMap > &results, Size dynamic_dim)
Concatenate the tensors in the ValueMap along the batch dimension.
std::map< VariableName, Tensor > ValueMap
Definition Tensor.h:39
int64_t Size
Definition types.h:71
ValueMap valuemap_move_device(ValueMap &&x, Device device)
Move all tensors in a ValueMap to a device.