OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Made alias_tensor serializable.

This commit is contained in:
Davis King 2015-12-06 11:43:38 -05:00
parent ad40ddd3a5
commit 83ecf1d94e
2 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
dlib/dnn/tensor.h
View File

@ -521,6 +521,30 @@ namespace dlib
alias_tensor_instance inst; alias_tensor_instance inst;
}; };
inline void serialize(const alias_tensor& item, std::ostream& out)
{
int version = 1;
serialize(version, out);
serialize(item.num_samples(), out);
serialize(item.k(), out);
serialize(item.nr(), out);
serialize(item.nc(), out);
}
inline void deserialize(alias_tensor& item, std::istream& in)
{
int version = 0;
deserialize(version, in);
if (version != 1)
throw serialization_error("Unexpected version found while deserializing dlib::alias_tensor.");
long num_samples, k, nr, nc;
deserialize(num_samples, in);
deserialize(k, in);
deserialize(nr, in);
deserialize(nc, in);
item = alias_tensor(num_samples, k, nr, nc);
}
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
} }

6
dlib/dnn/tensor_abstract.h
View File

@ -527,6 +527,12 @@ namespace dlib
!*/ !*/
}; };
void serialize(const alias_tensor& item, std::ostream& out)
void deserialize(alias_tensor& item, std::istream& in);
/*!
provides serialization support for alias_tensor.
!*/
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
} }