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