torch.Tensor.element_size

Tensor.element_size() → int

Возвращает размер в байтах отдельного элемента.

Пример:

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

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.Tensor.element_size.html