替代张量流中的np.insert

Question

numpy中有一个函数将给定值插入数组： https://docs.scipy.org/doc/numpy/reference/generated/numpy.insert.html

tensorflow中有类似的东西吗？

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

Answer 1

tf.nn.conv2d_transpose可以执行此上采样（仔细设计output_shape和strides）。示例代码：

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会对你有所帮助。