pytorch中的张量tensor数据的维度和数量查看

MAR-Sky

2030人浏览 · 2022-11-11 18:33:19

MAR-Sky · 2022-11-11 18:33:19 发布

查看tensor的数据分布情况

import torch

a = [[
		[1,2],
		[3,4]
	]]
b = torch.tensor(a)
print(b)
print(b.size()) #torch.Size([1, 2, 2])
print(b.shape)  #torch.Size([1, 2, 2])
print(b.ndim)  # 3

从结果看来，size()函数和shape属性返回的结果是一样的，属性ndim是返回维度的

乐奇 Rokid 开放社区

更多推荐

AI 智能眼镜的柔性神经，开启轻量化人机交互新世代

AI 智能眼镜的核心诉求，是在接近传统眼镜的轻薄体积内，塞进 AI 主控芯片、摄像头、麦克风阵列、骨传导 / 扬声器、光波导显示模组、触控传感器、电池、蓝牙 / Wi-Fi 射频模块等大量器件。随着 AI 眼镜出货量快速上行，鹏鼎、弘信、则成等国内 FPC 厂商纷纷布局 AI 眼镜软板、软硬结合板量产方案，单机 FPC 价值量显著高于 TWS 耳机，带动国内柔性线路板供应链升级，倒逼国内厂商突破超