PyTorch：torch.zeros_like()的用法

最新推荐文章于 2025-05-30 17:02:42 发布

原创最新推荐文章于 2025-05-30 17:02:42 发布 · 6.7k 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习

Pytorch 专栏收录该内容

19 篇文章

订阅专栏

本文介绍如何使用PyTorch中的torch.zeros_like()函数来生成与给定张量a具有相同形状的全零张量，通过实例演示了其用法及其在实际项目中的应用。

部署运行你感兴趣的模型镜像

作用：产生一个与a相同shape的Tensor.

举例：

import torch
a = torch.rand(3,4)  # 产生一个3行4列的0~1的随机Tensor
b = torch.zeros_like(a)  # 产生一个与a相同shape的全零Tensor
print('a:',a)
print('b:',b)


'''   运行结果   '''
a: tensor([[0.9285, 0.0966, 0.2180, 0.1234],
           [0.8930, 0.9143, 0.8272, 0.1228],
           [0.7441, 0.3232, 0.2867, 0.6598]])
b: tensor([[0., 0., 0., 0.],
           [0., 0., 0., 0.],
           [0., 0., 0., 0.]])

您可能感兴趣的与本文相关的镜像