labels.unsqueeze(1) 维度增加

labels.unsqueeze(1) 是 PyTorch 中的一个方法,用于在给定维度上增加一个维度。这个方法的作用是将原来的一维张量 labels 转换为一个二维张量,其中新增的维度位于指定位置上。

让我们更详细地解释一下这个方法的用法和意义:

假设原始的 labels 张量的形状是 [N],其中 N 是样本的数量,例如 [1, 2, 0, 1, 3]。在这种情况下,张量 labels 是一个一维张量,表示每个样本对应的标签。

现在,如果我们执行 labels.unsqueeze(1),它将在维度1的位置(索引位置从0开始)上增加一个新的维度。这将导致 labels 张量的形状从 [N] 变为 [N, 1]。结果是一个二维张量,其中有N行和1列,每一行表示一个样本的标签。

举例来说,假设 labels = [1, 2, 0, 1, 3],那么在执行 labels.unsqueeze(1) 后,结果将是:

[[1],
 [2],
 [0],
 [1]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

K同学啊

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值