Caffe2 - C++ API
A deep learning, cross platform ML framework
variable_tensor_list.h
1 #pragma once
2 #include <ATen/ATen.h>
3 
4 namespace torch {
5 namespace jit {
6 
7 // a wrapper to mark places where we expect all the at::Tensors to be
8 // variables
9 struct variable_tensor_list : public std::vector<at::Tensor> {
10  variable_tensor_list() = default;
11  template <class InputIt>
12  variable_tensor_list(InputIt first, InputIt last)
13  : std::vector<at::Tensor>(first, last) {}
14  explicit variable_tensor_list(std::vector<at::Tensor>&& tensor)
15  : std::vector<at::Tensor>(std::move(tensor)) {}
16 };
17 
18 } // namespace jit
19 } // namespace torch
Definition: jit_type.h:17