FM / data

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/api/include/torch/data/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/api/include/torch/data

Name	Size	Mode	Actions
dataloader/	-	0755	rm
datasets/	-	0755	rm
detail/	-	0755	rm
samplers/	-	0755	rm
transforms/	-	0755	rm
dataloader.h	2166	0644	edit dl rm
dataloader_options.h	2207	0644	edit dl rm
datasets.h	289	0644	edit dl rm
example.h	1308	0644	edit dl rm
iterator.h	5284	0644	edit dl rm
samplers.h	318	0644	edit dl rm
transforms.h	222	0644	edit dl rm
worker_exception.h	1140	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/api/include/torch/data/example.h (1308B)

#pragma once

#include <torch/types.h>

namespace torch {
namespace data {

/// An `Example` from a dataset.
///
/// A dataset consists of data and an associated target (label).
template <typename Data = Tensor, typename Target = Tensor>
struct Example {
  using DataType = Data;
  using TargetType = Target;

Example() = default;
  Example(Data data, Target target)
      : data(std::move(data)), target(std::move(target)) {}

Data data;
  Target target;
};

namespace example {
using NoTarget = void;
} //  namespace example

/// A specialization for `Example` that does not have have a target.
///
/// This class exists so that code can be written for a templated `Example`
/// type, and work both for labeled and unlabeled datasets.
template <typename Data>
struct Example<Data, example::NoTarget> {
  using DataType = Data;
  using TargetType = example::NoTarget;

Example() = default;
  /* implicit */ Example(Data data) : data(std::move(data)) {}

// When a DataLoader returns an Example like this, that example should be
  // implicitly convertible to the underlying data type.

operator Data&() {
    return data;
  }
  operator const Data&() const {
    return data;
  }

Data data;
};

using TensorExample = Example<Tensor, example::NoTarget>;
} // namespace data
} // namespace torch

#pragma once #include namespace torch { namespace data { /// An `Example` from a dataset. /// /// A dataset consists of data and an associated target (label). template struct Example { using DataType = Data; using TargetType = Target; Example() = default; Example(Data data, Target target) : data(std::move(data)), target(std::move(target)) {} Data data; Target target; }; namespace example { using NoTarget = void; } // namespace example /// A specialization for `Example` that does not have have a target. /// /// This class exists so that code can be written for a templated `Example` /// type, and work both for labeled and unlabeled datasets. template struct Example { using DataType = Data; using TargetType = example::NoTarget; Example() = default; /* implicit */ Example(Data data) : data(std::move(data)) {} // When a DataLoader returns an Example like this, that example should be // implicitly convertible to the underlying data type. operator Data&() { return data; } operator const Data&() const { return data; } Data data; }; using TensorExample = Example; } // namespace data } // namespace torch