1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
| #pragma once
|
| #include <torch/serialize/archive.h>
| #include <torch/types.h>
|
| namespace torch {
| inline serialize::OutputArchive& operator<<(
| serialize::OutputArchive& archive,
| const Tensor& tensor) {
| archive.write("0", tensor);
| return archive;
| }
|
| inline serialize::InputArchive& operator>>(
| serialize::InputArchive& archive,
| Tensor& tensor) {
| archive.read("0", tensor);
| return archive;
| }
| } // namespace torch
|
|
