torch.Tensor.nbytes

Tensor.nbytes

Возвращает количество байтов, занимаемых «представлением» элементов тензора, если в тензоре не используется разреженный формат хранения. Определяется как numel() * element_size()