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

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

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

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

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

执行后的结果为:

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

相关推荐
运器1232 分钟前
【一起来学AI大模型】微调技术:LoRA(Low-Rank Adaptation) 的实战应用
人工智能·pytorch·python·深度学习·ai·ai编程
maomaolaoshi9 分钟前
【读书笔记】从AI到Transformer:LLM技术演进全解析
人工智能·深度学习·transformer
天地一流殇9 分钟前
深度学习-词嵌入
人工智能·深度学习
AIGC包拥它16 分钟前
AI教学设计助手:生成好教案的Prompt技术实战(二)
人工智能·prompt·aigc
Ronin-Lotus2 小时前
模型训练与部署注意事项篇---resize
人工智能·深度学习·计算机视觉
我爱一条柴ya3 小时前
【AI大模型】LLM模型架构深度解析:BERT vs. GPT vs. T5
人工智能·python·ai·ai编程
Coovally AI模型快速验证3 小时前
从FCOS3D到PGD:看深度估计如何快速搭建你的3D检测项目
人工智能·深度学习·神经网络·yolo·3d·cnn
屁股割了还要学7 小时前
快速过一遍Python基础语法
开发语言·python·学习·青少年编程
kikikidult7 小时前
Ubuntu20.04运行openmvg和openmvs实现三维重建(未成功,仅供参考)
人工智能·笔记·ubuntu·计算机视觉