TensorMetadata

class torch.fx.passes.shape_prop.TensorMetadata(shape, dtype, requires_grad, stride, memory_format, is_quantized, qparams) [исходный код]

Структура, содержащая важную информацию о тензоре в программе PyTorch.

Примечание

Совместимость этого API с предыдущими версиями гарантируется.

count(value, /)

Возвращает количество вхождений значения.

dtype: dtype

Псевдоним для поля номер 1

index(value, start=0, stop=9223372036854775807, /)

Возвращает первый индекс значения.

Вызывает ValueError, если значение отсутствует.

is_quantized: bool

Псевдоним для поля номер 5

memory_format: memory_format | None

Псевдоним для поля номер 4

qparams: dict[str, Any]

Псевдоним для поля номер 6

requires_grad: bool

Псевдоним для поля номер 2

shape: Size

Псевдоним для поля номер 0

stride: tuple[int, ...]

Псевдоним для поля номер 3