Returns a vector of size 'nodes' with each node being of size 'node_size'.
| 55 | |
| 56 | // Returns a vector of size 'nodes' with each node being of size 'node_size'. |
| 57 | std::vector<Node*> VarVector(Graph* const g, const int nodes, |
| 58 | const int node_size) { |
| 59 | std::vector<Node*> result; |
| 60 | result.reserve(nodes); |
| 61 | for (int i = 0; i < nodes; ++i) { |
| 62 | result.push_back(Var(g, node_size)); |
| 63 | } |
| 64 | return result; |
| 65 | } |
| 66 | |
| 67 | Node* Zeros(Graph* const g, const TensorShape& shape) { |
| 68 | Tensor data(DT_FLOAT, shape); |