Fonction torch.empty_strided de PyTorch


Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.empty_stridedC'est une fonction dans PyTorch pour créer un tenseur non initialisé avec un stride spécifié.

Définition de fonction

torch.empty_strided(size, stride, dtype=None, device=None, requires_grad=False)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur de forme (3, 4) avec un stride (1, 3)
x = torch.empty_strided((3, 4), (1, 3))

print(x)
print("Forme :", x.shape)
print("Stride:", x.stride())
print("Remarque : les valeurs sont des valeurs de mémoire non initialisées")

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions