mirror of https://github.com/davisking/dlib.git
Added is_vector() for tensor objects.
This commit is contained in:
parent
fa8b3767bf
commit
2a2956a238
|
@ -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 (
|
||||
|
|
|
@ -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 (
|
||||
|
|
Loading…
Reference in New Issue