欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

pytorch实现列向量变矩阵

程序员文章站 2024-03-16 18:58:52
...
import torch
x = torch.Tensor(3,1)
print(x)
y = torch.Tensor(3,3)
print(y.copy_(x))
print(id(x), id(y))

代码运行如下:
pytorch实现列向量变矩阵

行数不同的但是元素个数相同:

import torch
x = torch.Tensor(4).fill_(1)
y = torch.Tensor(2,2).copy_(x) #也可以实现不同Tensor的复制。
print(x)
print(y)

pytorch实现列向量变矩阵

借鉴的地址:
https://blog.csdn.net/hungryof/article/details/51802829 中tensor的复制

相关标签: 单列变矩阵