#pragma once #include #include #include #include // Some "exports". namespace torch { namespace data { using datasets::BatchDataset; using datasets::Dataset; } // namespace data } // namespace torch