pytorch之torch.zeros_like,torch.ones_like和tensor按索引赋值

1. torch.zeros_like

可以生成和括号内变量维度一致的全是0的内容。例如:输入维度为 1 × 8 × 3 × 5 1\times8\times3\times5 1×8×3×5的x,要生成跟x相同维度且值全为0的y,则如下所示:

在这里插入图片描述

2. torch.ones_like

可以生成和括号内变量维度一致的全是1的内容。例如:输入维度为 1 × 8 × 3 × 5 1\times8\times3\times5 1×8×3×5的x,要生成跟x相同维度且值全为1的y,则如下所示:
在这里插入图片描述

3. tensor按索引赋值

猜你喜欢

转载自blog.csdn.net/weixin_43994864/article/details/126765769