Skip to content

Commit 6bcf470

Browse files
authored
add DataChunkFlat constructors with the vector initialization
1 parent 24a7d53 commit 6bcf470

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

keras_model.h

+5
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,12 @@ class keras::DataChunk2D : public keras::DataChunk {
7777

7878
class keras::DataChunkFlat : public keras::DataChunk {
7979
public:
80+
DataChunkFlat(size_t size) : f(size) { }
81+
DataChunkFlat(size_t size, float init) : f(size, init) { }
82+
DataChunkFlat(void) { }
83+
8084
std::vector<float> f;
85+
std::vector<float> & get_1d_rw() { return f; }
8186
std::vector<float> const & get_1d() const { return f; }
8287
void set_data(std::vector<float> const & d) { f = d; };
8388
size_t get_data_dim(void) const { return 1; }

0 commit comments

Comments
 (0)