pytorch之我的沙比错误

编程入门 行业动态 更新时间:2024-10-09 22:23:51

pytorch之我的沙比<a href=https://www.elefans.com/category/jswz/34/1771449.html style=错误"/>

pytorch之我的沙比错误

矩阵赋值

描述: 有batch维度的矩阵赋值
思路: 先生成需要的全零矩阵,由于我这每个batch里的值都一样,因此就在第零维全部选择,第一维选择具体的位置,第二维选择区间。

y = torch.zeros(2, 2, 4)
c = torch.zeros(2, 2, 4)
for i in range(2):y[0][i][2*i:2*i+2] = 1.y[1][i][2*i:2*i+2] = 1.c[0:2][i][2*i:2*i+2] = 1.
print(y)
print(c)

愚蠢描述:按照思路,c和y矩阵应该相同,但是并不。c[0:2]之后仍然是个(2,2,4),因此之后的[i]选择的仍是第0维。


更多推荐

pytorch之我的沙比错误

本文发布于:2024-03-13 00:16:00,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1732751.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:错误   pytorch

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!