Атрибут ndim
Атрибут ndim класса Tensor возвращает количество измерений тензора.
Это целое число, которое показывает, сколько осей имеет тензор.
Атрибут доступен только для чтения, его нельзя изменить напрямую.
Синтаксис
tensor.ndim
Пример
Давайте создадим скаляр (тензор нулевой размерности) и проверим его атрибут ndim:
import torch
t = torch.tensor(5)
res = t.ndim
print(res)
Результат выполнения кода:
0
Пример
Давайте создадим одномерный тензор и проверим его размерность:
import torch
t = torch.tensor([1, 2, 3, 4, 5])
res = t.ndim
print(res)
Результат выполнения кода:
1
Пример
Давайте создадим двумерный тензор и проверим его размерность:
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
])
res = t.ndim
print(res)
Результат выполнения кода:
2
Пример
Давайте создадим трехмерный тензор и проверим его размерность:
import torch
t = torch.tensor([
[
[1, 2],
[3, 4],
],
[
[5, 6],
[7, 8],
],
])
res = t.ndim
print(res)
Результат выполнения кода:
3
Пример
Давайте создадим четырехмерный тензор и проверим его размерность:
import torch
t = torch.zeros(2, 3, 4, 5)
res = t.ndim
print(res)
Результат выполнения кода:
4
Пример
Давайте сравним атрибут ndim с методом dim, который возвращает аналогичное значение:
import torch
t = torch.tensor([
[1, 2, 3],
[4, 5, 6],
])
print(t.ndim)
print(t.dim())
Результат выполнения кода:
2
2
