提交 ea633907 编写于 作者: H helinwang 提交者: GitHub

Merge pull request #5091 from helinwang/init

Fix new pserver having random momemtum initial values
...@@ -15,7 +15,8 @@ template <class T> ...@@ -15,7 +15,8 @@ template <class T>
class TensorT { class TensorT {
public: public:
TensorT(size_t size) : height_(1), width_(size) { TensorT(size_t size) : height_(1), width_(size) {
data_ptr_ = std::shared_ptr<T>(new T[size], std::default_delete<T[]>()); // new T[size]() initializes all element to zero value.
data_ptr_ = std::shared_ptr<T>(new T[size](), std::default_delete<T[]>());
data_ = data_ptr_.get(); data_ = data_ptr_.get();
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册