列表（list）、数组（numpy）、张量（tensor）、元组各自怎样查看形状

列表（list）、数组（numpy）、张量（tensor）、元组各自怎样查看形状

一、查看列表（list）的形状

1.shape

import numpy as np

L = [[1,2],[3,4],[5,6],[7,8]]
L = np.array(L)

print(L.shape)

二、查看数组（numpy）的形状

1.shape

a=np.array([[1,2,3],[4,5,6]],dtype=np.float)
print(a.shape)

三、查看张量（tensor）的形状（dim,shape,size,numel）

1. torch.Tensor.dim
   dim的作用是返回张量的维数

import torch
a = torch.rand((3,4))  #生成随机数
a.dim()

2.torch.Tensor.shape
shape的作用是返回张量的形状

import torch
a = torch.rand((3,4))
a.shape

3.torch.Tensor.size
shape的作用是返回张量的形状，可指定dim返回某一维度的大小

import torch
a = torch.rand((2,3,4))
print(a.size())
print(a.size(dim=0))  #维度的指标是从0开始
print(a.size(dim=2))

4.torch.Tensor.numel
numel的作用是返回张量中元素的数量

import torch
a = torch.rand((2,3,4))
print(a.numel())  #2×3×4=24
#除此之外还可以使用torch.numel()
print(torch.numel(a))

四、元组