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维。

---