Python PyTorch Tensor.element_size用法及代码示例

本文简要介绍python语言中 torch.Tensor.element_size 的用法。

用法: Tensor.element_size() → int

返回单个元素的大小(以字节为单位)。

例子：

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

相关用法

注：本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.Tensor.element_size。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。