torch.Tensor.element_size
- Tensor.element_size() → int
-
返回单个元素的字节大小。
示例：
>>> torch.tensor([]).element_size() 4 >>> torch.tensor([], dtype=torch.uint8).element_size() 1
本页目录
•
pytorch-doc-2.5中文文档
返回单个元素的字节大小。
示例：
>>> torch.tensor([]).element_size() 4 >>> torch.tensor([], dtype=torch.uint8).element_size() 1