pytorch相关小操作的总结

本文介绍了PyTorch中的expand_as()函数,通过示例展示了如何使用该函数将一个张量的维度扩充至另一个张量的维度,以便进行广播操作。在示例中,张量a从(4,1)扩充到(4,2),与张量c的维度匹配,但原始张量a的维度并未改变。这个操作在处理多维数据时非常有用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档


前言

这里主要记录自己python甚至 pytorch的学习内容


一、expand_as()的理解

这是个pytorch中的操作,这里

示例:pandas 是基于NumPy 的一种工具,该工具是为了解决数据分析任务而创建的。

from __future__ import print_function
import torch  as t

# a = t.Tensor(2,3)
#
# print(a)

# b = t.Tensor([[1,2,3],[4,5,6]]) # 用list的数据创建tensor
#
# print(b)
#
# print(b.size())
#
# print(b.numel())
#
# print(t.__version__)


a = t.tensor([[2],[3],[4],[5]])
c = t.tensor([[2,2],[3,3],[5,5],[6,6]]).view(4,2)
print(c)

d = a.expand_as(c)

print(d)

print(a)


首先可以看出a的维度是(4,1)

c的维度是(4,2)

a.expand_as©的语句的意思就是将a的维度扩充为c的维度

tensor([[2, 2],
        [3, 3],
        [5, 5],
        [6, 6]])
tensor([[2, 2],
        [3, 3],
        [4, 4],
        [5, 5]])
tensor([[2],
        [3],
        [4],
        [5]])

print(d)就是a扩充维度之后的输出,但是a本身的维度没有变化。
OK,got it。

二、XX

1.XX

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

夜夜夜Best

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

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

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

打赏作者

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

抵扣说明:

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

余额充值