3 #include <torch/data/example.h> 4 #include <torch/data/transforms/base.h> 5 #include <torch/types.h> 12 namespace transforms {
17 template <
typename Target = Tensor>
19 :
public Transform<Example<Tensor, Target>, Example<Tensor, Target>> {
22 using typename Transform<E, E>::InputType;
30 input.data = (*this)(std::move(input.data));
36 template <
typename Target = Tensor>
39 using FunctionType = std::function<Tensor(Tensor)>;
43 : function_(
std::move(function)) {}
47 return function_(std::move(input));
51 FunctionType function_;
56 template <
typename Target = Tensor>
70 return input.sub(mean).div(stddev);
An Example from a dataset.
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory)...