Commit 83ecf1d9 authored by Davis King's avatar Davis King

Made alias_tensor serializable.

parent ad40ddd3
...@@ -521,6 +521,30 @@ namespace dlib ...@@ -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);
}
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
} }
......
...@@ -527,6 +527,12 @@ namespace dlib ...@@ -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.
!*/
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment