A = torch.arange(0, 16)
print(A)
B = A.view(-1,2)
print(B)
C = A.view(-1,2,2)
print(C)

本文通过实例展示了如何使用PyTorch进行张量创建、查看和转换操作,包括一维、二维和三维张量的创建及视图转换。
A = torch.arange(0, 16)
print(A)
B = A.view(-1,2)
print(B)
C = A.view(-1,2,2)
print(C)


被折叠的 条评论
为什么被折叠?