В чем разница между Tensor.size и Tensor.shape в PyTorch? - PullRequest
0 голосов
/ 05 августа 2020

В чем разница между Tensor.size и Tensor.shape в Pytorch? Я хочу получить количество элементов и размеры Tensor. Например, для тензора с размерами 2 на 3 на 4 я ожидаю 24 для количества элементов и (2,3,4) для измерения. Спасибо.

1 Ответ

2 голосов
/ 05 августа 2020

.shape является псевдонимом для .size() и был добавлен для более точного соответствия numpy, см. Это обсуждение здесь .

...