官方文档指路:
torch.zeros.like()
该语句的主要作用是生成一个值全为0的、维度与输入尺寸相同的矩阵。
torch.zeros_like(input)相当于torch.zeros(input.size(),dtype=input.dtype,layout=input.layout, device=input.device)
input = torch.empty(2,3)
torch.zeros_like(input)
结果为:
官方文档指路:
torch.zeros.like()
该语句的主要作用是生成一个值全为0的、维度与输入尺寸相同的矩阵。
torch.zeros_like(input)相当于torch.zeros(input.size(),dtype=input.dtype,layout=input.layout, device=input.device)
input = torch.empty(2,3)
torch.zeros_like(input)
结果为: