torch.Tensor.element_size

Tensor.element_size() → int

傳回單個元素的大小（以位元組為單位）。

範例

>>> torch.tensor([]).element_size()
4
>>> torch.tensor([], dtype=torch.uint8).element_size()
1