如何重塑张量并获得 Tensorflow 中的第一个维度？

Question

我有一个张量。 假设它的维度是 [2, 999]。 如何将其重塑为 [999, 2] 并获得第一个维度，即 Tensorflow 中的 999？

Answer 1

重塑：

new_tensor = tensor.reshape((999,2))

查找第一个维度：

first_dimension = tensor.shape[0]

Answer 2

您好，您可以这样做：

顺便说一句，张量对象没有重塑 function 你必须使用 tf.reshape function 来调用它

import tensorflow as tf 
tensor = tf.range(999 * 2) # create a tensor
reshaped_tensor = tf.reshape(tensor, (2, 999))

# this doesn't work
tensor = tensor.reshape(999, 2)

您可以使用转置命令切换轴，因为您具有相同的尺寸

switched_axis = tf.transpose(tensor) # assuming tensor has a shape of 999, 2

打印第一个尺寸

print(tensor.shape[0])