2017-11-29 33 views
0

xのブロック更新形状のテンソルは[32,32,3]Tensorflow - テンソル

Yは、我々はY、Yを[更新することができる形状【1,320,320,3]

のテンソルであります0、:32、:32、:] = xをテンソルフロー演算として?

答えて

0

はい、できます。あなたはそれをまったくテストしましたか?

スライスもテンソルです。用途:1Dで

Y[0,:32,:32,:].assign(x) 

例我々は明確に何が起こっているかを見ることができるように:

import numpy as np 
import tensorflow as tf 


x = tf.Variable(np.zeros((2,))) 
Y = tf.Variable(np.ones((10,))) 

block_update = Y[4:6].assign(x) 

with tf.Session() as sess: 
    sess.run(tf.global_variables_initializer()) 

    print('old', Y.eval(sess)) 
    sess.run(block_update) 
    print('new', Y.eval(sess)) 

出力:

old [ 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.] 
new [ 1. 1. 1. 1. 0. 0. 1. 1. 1. 1.] 
関連する問題