PyTorch torch.zeros_like 函数
Pytorch torch 参考手册torch.zeros_like 是 PyTorch 中用于创建与输入张量形状相同的全零张量的函数。
函数定义
torch.zeros_like(input, dtype=None, device=None, requires_grad=False)
使用示例
实例
import torch
x = torch.randn(3, 4)
y = torch.zeros_like(x)
print("x 形状:", x.shape)
print("y 形状:", y.shape)
print(y)
x = torch.randn(3, 4)
y = torch.zeros_like(x)
print("x 形状:", x.shape)
print("y 形状:", y.shape)
print(y)