результат свертки тензорного потока к numy

Question

Я пишу простой код,

import numpy as np
import tensorflow as tf

x_data = np.loadtxt('D:\proj\dnn_lib_cuda\input')
w_data = np.loadtxt('D:\proj\dnn_lib_cuda\weight')

x_tensor = np.reshape(x_data, (1, 3, 224, 224))
w_tensor = np.reshape(w_data, (64, 3, 3, 3))

x_tensor_ch = x_tensor.transpose(0, 2, 3, 1)
w_tensor_ch = w_tensor.transpose(2, 3, 1, 0)


x = tf.placeholder(tf.float32, shape = (1, 224, 224, 3))
w = tf.placeholder(tf.float32, shape = (3, 3, 3, 64))

result = tf.nn.conv2d(input = x, filter = w, strides = [1, 1, 1, 1], padding = 'SAME')
sess = tf.Session()
sess.run(result, feed_dict = {x: x_tensor_ch, w:w_tensor_ch})
print(result)

Теперь результатом является тензор с формой (1, 224, 224, 64), как я могу получить данные в формате numpy?

Answer 1

sess.run(...) возвращает результат оценки тензора result с учетом данных, переданных в feed_dict.

Итак, вы хотите

output = sess.run(result, feed_dict = {x: x_tensor_ch, w:w_tensor_ch})