pytorch的一些基本操作

(1)生成一个未初始化的tensor

1 import torch
2 
3 x = torch.Tensor(5,3)
4 print(x)

(2)随机初始化一个tensor

y = torch.randn(5,3)
print(y)

 (3)矩阵的大小,直接将其当做元组来操作

print(y.size())

(4)tensor之间的操作,可以有多重形式的操作,比如相加操作,直接+,或者add,等等

print(torch.add(x, y))

print(x+y)

(5)tensor和numpy之间的呼唤,也是非常方便,除了chartensor之外,所有的tensor都能够转换为numpy

y_n = y.numpy()
print(y_n)
y_t = torch.from_numpy(y_n)
print(y_t)

 未完待续

原文地址:https://www.cnblogs.com/demo-deng/p/10598286.html