tensor如何插入一行为0或者一列为0 其他类似，实现append
这些曲线真难
·
import torch
t1 = torch.tensor([[1,6,4],[3,4,5]])
#在第二列后插入一列与第二列一样
t1=t1[:,[0,1,1,2]]
t1[:,2]=0 #第三列为0
print(t1)
#---------显示为：------------
tensor([[1, 6, 0, 4],
[3, 4, 0, 5]])
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)