2017-08-23 50 views
0

有在numpy的功能进行的插入给定的值到数组np.insert?替代在tensorflow

或者,张量流中是否有一个函数可以在张量值之间使用零进行张量上采样?

回答

0

tf.nn.conv2d_transpose可以做这种上采样(仔细设计output_shapestrides)。示例代码:

import tensorflow as tf 
import numpy as np 

input = tf.convert_to_tensor(np.ones((1, 20, 20, 1))) 
input = tf.cast(input, tf.float32) 
b = np.zeros((3, 3, 1, 1)) 
b[1, 1, 0, 0] = 1 
weight = tf.convert_to_tensor(b) 
weight = tf.cast(weight, tf.float32) 
output = tf.nn.conv2d_transpose(input, weight, output_shape=(1, 40, 40, 1), strides=[1, 2, 2, 1]) 

sess = tf.Session() 

print sess.run(output[0, :, :, 0]) 

我相信检查它的api会帮助你更多。