Tensor对象的属性函数
Tensor对象的属性函数
创建Tensor后，我们还需要看到它的一些属性，判断是否跟预期相符。注意Libtorch的Tensor是没有公开可访问的属性attribute的，Tensor信息需要属性函数来获取。常见的属性函数包括:
- dim(): Tensor的维度
- sizes(): 跟Pytorch中的shape属性一样
- size(n): 第N个维度的shape
- numel(): 总的元素数目，sizes中的每个元素相乘
- dtype(): 数据类型
- device(): Tensor所在的设备类型，CPU, CUDA, MPS等。
使用方式如下:
// Tensor 属性函数
torch::Tensor foo = torch::randn({1, 3, 224, 224});
auto dim = foo.dim(); // 4
auto sizes = foo.sizes(); // [1, 3, 224, 224]
auto size_0 = foo.size(0); // 1
auto numel = foo.numel(); // 150528
auto dtype = foo.dtype(); // float
auto scalar_type = foo.scalar_type(); // Float
auto device = foo.device(); // cpu网站声明：如果转载，请联系本站管理员。否则一切后果自行承担。
赞同 0
评论 0 条
字体处理常用属性