Función PyTorch torch.empty_strided


Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.empty_stridedEs una función en PyTorch que se utiliza para crear tensores no inicializados con un stride (paso) especificado.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor de forma (3, 4), con stride (1, 3)
x = torch.empty_strided((3, 4), (1, 3))

print(x)
print(Forma:, x.shape)
print(Stride:, x.stride())
print(Nota: los valores son valores de memoria no inicializados)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones