在PyTorch中,对于一个张量,如何快速为多个元素赋值相同的值

我们以"a = torch.arange(12).reshape((3, -1))"为例,a里面现在是:

如果我们想让a的右下角的2行3列的元素都为10的话,可以如何快速实现呢?

我们可以用到索引和切片技术,执行如下的指令即可达到目标:

python 复制代码
a[1:3, 1:] = 10

执行后的结果为:

这样,我们就达到了预期。

相关推荐
weixin_470740365 小时前
某算法的python执行汇编
汇编·python·算法
OAFD.5 小时前
机器学习之线性回归:原理、实现与实践
人工智能·机器学习·线性回归
SHIPKING3937 小时前
【机器学习&深度学习】LMDeploy的分布式推理实现
人工智能·深度学习
mit6.8247 小时前
[RestGPT] docs | RestBench评估 | 配置与环境
人工智能·python
CareyWYR8 小时前
每周AI论文速递(250818-250822)
人工智能
门思科技8 小时前
LoRaWAN 的网络拓扑全解析:架构、原理与应用实践
服务器·网络·人工智能·科技·物联网·架构
兔子的倔强8 小时前
Transformer在文本、图像和点云数据中的应用——经典工作梳理
人工智能·深度学习·transformer
Ice__Cai8 小时前
Flask 之 Cookie & Session 详解:用户状态管理
后端·python·flask·cookie·session
lxmyzzs9 小时前
【图像算法 - 21】慧眼识虫:基于深度学习与OpenCV的农田害虫智能识别系统
人工智能·深度学习·opencv·算法·yolo·目标检测·计算机视觉