out Class — pytorch Architecture

Source Code

aten/src/ATen/core/tensor_type.cpp lines 65–86

template <typename T>
std::ostream& operator<<(std::ostream& out, const VaryingShape<T>& vs) {
  out << '(';
  if (!vs.size()) {
    out << "*)";
    return out;
  }

  for (size_t i = 0; i < vs.size(); i++) {
    if (i > 0) {
      out << ", ";
    }
    auto const& v = vs[i];
    if (v.has_value()) {
      out << v.value();
    } else {
      out << '*';
    }
  }
  out << ')';
  return out;
}