如何查看一个变量的维度

在Python中，我们可以使用numpy库的shape属性来查看变量的维度。如果变量是一个numpy数组，我们可以直接使用变量名.shape来查看其维度，例如：

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
print(a.shape)  # 输出 (2, 3)

如果变量不是一个numpy数组，我们可以先将其转换为numpy数组，再使用shape属性查看其维度，例如：

import numpy as np

a = [[1, 2, 3], [4, 5, 6]]
a = np.array(a)
print(a.shape)  # 输出 (2, 3)

如果变量是一个PyTorch张量，我们可以使用其size()方法来查看其维度，例如：

import torch

a = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(a.size())  # 输出 torch.Size([2, 3])

如果变量是一个TensorFlow张量，我们可以使用其shape属性来查看其维度，例如：

import tensorflow as tf

a = tf.constant([[1, 2, 3], [4, 5, 6]])
print(a.shape)  # 输出 (2, 3)