2
を参照してください。Best way to flatten a 2D tensor containing a vector in TensorFlow?私は2DテンソルをTensorFlowの1Dテンソルにフラット化する方法を見ていますが、 TensorFlow機能を備えた3Dテンソルへのテンソル?例えば形状(n、h、w、c)から(n、h、w、c)への形状変更
:
import tensorflow as tf
X = tf.random_normal([n, h, w, c], mean=1, stddev=4)
Y = ???