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

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

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

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

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

执行后的结果为:

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

相关推荐
liuyunshengsir37 分钟前
神经网络中的梯度的计算详解
人工智能·大模型
XINVRY-FPGA40 分钟前
XCVU47P-2FSVH2892E Xilinx Virtex UltraScale+ FPGA AMD
c语言·c++·人工智能·嵌入式硬件·阿里云·fpga开发·fpga
___波子 Pro Max.2 小时前
python list去重
python·list
Ai财富密码3 小时前
机器学习 (ML) 基础入门指南
人工智能·神经网络·机器学习·机器人·ml
华科易迅3 小时前
人工智能学习38-VGG训练
人工智能·学习·人工智能学习38-vgg训练
狐凄4 小时前
Python实例题:基于边缘计算的智能物联网系统
python·物联网·边缘计算
m0_537437574 小时前
【深度学习基础与概念】笔记(一)深度学习革命
人工智能·笔记·深度学习
@十八子德月生4 小时前
第十章——8天Python从入门到精通【itheima】-99~101-Python基础综合案例-数据可视化(案例介绍=JSON格式+pyecharts简介)
大数据·python·信息可视化·pycharm·echarts·数据可视化
W说编程4 小时前
算法导论第二十四章 深度学习前沿:从序列建模到创造式AI
c语言·人工智能·python·深度学习·算法·性能优化
动能小子ohhh5 小时前
html实现登录与注册功能案例(不写死且只使用js)
开发语言·前端·javascript·python·html