记录张量Tensor的属性,以及获取 tensor 形状的方法

属性:

形状(shape):Tensor的shape,是一个tuple

维数(ndim):Tensor的秩,也就是len(tensor.shape),是一个整数

元素个数(size):Tensor中所有元素的个数,是一个整数

单个元素大小(itemsize):Tensor中每一个元素占用字节数,是一个整数

数据类型(dtype):Tensor的dtype,是MindSpore的一个数据类型

占用字节数量(nbytes):Tensor占用的总字节数,是一个整数

每一维步长(strides):Tensor每一维所需要的字节数,是一个tuple

转置张量(T):Tensor的转置,是一个Tensor

pycharm 获取 tensor 形状的方法有两种:shape 和 size()

import torch

a = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(a.shape)
print(a.size())

你可能感兴趣的:(python)