PyTorch torch.empty_strided 函数
Pytorch torch 参考手册torch.empty_strided 是 PyTorch 中用于创建具有指定步幅的未初始化张量的函数。
函数定义
torch.empty_strided(size, stride, dtype=None, device=None, requires_grad=False)
使用示例
实例
import torch
# 创建形状为 (3, 4) 的张量，步幅为 (1, 3)
x = torch.empty_strided((3, 4), (1, 3))
print(x)
print("形状:", x.shape)
print("步幅:", x.stride())
print("注意: 值为未初始化的内存值")
# 创建形状为 (3, 4) 的张量，步幅为 (1, 3)
x = torch.empty_strided((3, 4), (1, 3))
print(x)
print("形状:", x.shape)
print("步幅:", x.stride())
print("注意: 值为未初始化的内存值")