如何查看tensor的维度使用什么指令

在TensorFlow或PyTorch等深度学习框架中，你可以使用特定的函数来获取张量（tensor）的维度。以下是各自语言的示例：

**TensorFlow (TF)**:

import tensorflow as tf

# 假设 tensor variable 是一个张量
tensor_variable = tf.constant([[1, 2, 3], [4, 5, 6]])

# 使用shape属性获取维度
tensor_shape = tensor_variable.shape
print("Tensor dimensions:", tensor_shape)

**PyTorch**:

import torch

# 假设 tensor 是一个张量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 使用size() 或者 dim() 函数获取维度
tensor_dim = tensor.dim()
print("Tensor dimensions:", tensor_dim)