1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
| #pragma once
|
| #include <c10/core/Scalar.h>
| #include <ATen/core/Tensor.h>
| #include <iostream>
|
|
| namespace c10 {
| CAFFE2_API std::ostream& operator<<(std::ostream& out, Backend b);
| }
| namespace at {
|
| CAFFE2_API std::ostream& operator<<(std::ostream& out, const DeprecatedTypeProperties& t);
| CAFFE2_API std::ostream& print(
| std::ostream& stream,
| const Tensor& tensor,
| int64_t linesize);
| static inline std::ostream& operator<<(std::ostream & out, const Tensor & t) {
| return print(out,t,80);
| }
| static inline void print(const Tensor & t, int64_t linesize=80) {
| print(std::cout,t,linesize);
| }
|
| static inline std::ostream& operator<<(std::ostream & out, Scalar s) {
| return out << (s.isFloatingPoint() ? s.toDouble() : s.toLong());
| }
|
| }
|
|
