分类：numel函数

2021
07-01

pytorch中的numel函数用法说明

获取tensor中一共包含多少个元素importtorchx=torch.randn(3,3)print("numberelementsofxis",x.numel())y=torch.randn(3,10,5)print("numberelementsofyis",y.numel())输出：numberelementsofxis9numberelementsofyis15027和150分别位x和y中各有多少个元素或变量补充：pytorch获取张量元素个数numel()的用法numel就是"numberofelements"的简写。numel()可以直接返回int类型的元素个数importtorcha=... 继续阅读 >