NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added is_vector() for tensor objects.

Browse Source
This commit is contained in:
Davis King 2016-10-26 16:49:56 -04:00
parent fa8b3767bf
commit 2a2956a238
2 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dlib/dnn/tensor.h
View File

@ -191,6 +191,18 @@ namespace dlib
long m_size; // always equal to m_n*m_k*m_nr*m_nc
};
// ----------------------------------------------------------------------------------------
inline bool is_vector (
const tensor& t
)
{
return t.size() == t.num_samples() ||
t.size() == t.k() ||
t.size() == t.nr() ||
t.size() == t.nc();
}
// ----------------------------------------------------------------------------------------
inline const matrix_op<op_pointer_to_mat<float> > mat (

14
dlib/dnn/tensor_abstract.h
View File

@ -345,6 +345,20 @@ namespace dlib
- This function blocks until the copy has completed.
!*/
// ----------------------------------------------------------------------------------------
bool is_vector (
const tensor& t
);
/*!
ensures
- returns true if and only if one of the following is true:
- t.size() == t.num_samples()
- t.size() == t.k()
- t.size() == t.nr()
- t.size() == t.nc()
!*/
// ----------------------------------------------------------------------------------------
const matrix_exp mat (