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