Skip to content
Snippets Groups Projects

Updates required for QAT

Merged Olivier BICHLER requested to merge qat into dev
Files
12
@@ -312,6 +312,18 @@ class Tensor : public Data,
@@ -312,6 +312,18 @@ class Tensor : public Data,
*/
*/
Tensor sqrt() const;
Tensor sqrt() const;
 
/**
 
* @brief Element-wise abs operation for Tensor.
 
* @return Tensor
 
*/
 
Tensor abs() const;
 
 
/**
 
* @brief Mean operation for Tensor.
 
* @return Tensor
 
*/
 
Tensor mean() const;
 
~Tensor() noexcept;
~Tensor() noexcept;
public:
public:
Loading