Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- for h{
- for w{
- for i in range 3{
- for j in range 3{
- new_tensor(i,j) = m_tensor(h+i,w+j)
- }
- }
- out = sess->run(out_node, in:new_tensor)
- m_tensor(h,w) = out.xxx
- }
- }
- std::copy_n(vec.begin(), vec.size(), input.flat<float>().data())
- std::memcpy(input.flat<float>().data(), vec.data(), input.NumElements() * sizeof(float))
- for i in range 3{
- for j in range 3{
- new_tensor(i,j) = m_tensor(h+i,w+j)
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement