(Pytorch)向量简单操作

Pytorch函数

向量

创建随机向量-randn()

randn()、is_tensor()、is_storage、size()、numel()

import torch as T

y = T.randn(1, 2, 3)
print(y)
# is_tensor判断是否为pytorch张量
print(T.is_tensor(y))
# is_storage判断是否为pytorch存储空间
print(T.is_storage(y))
# size可以查看创建向量维数
print(y.size())
# numel查看向量中所有元素个数
print(T.numel(y))

运行结果：

创建零向量-zeros()

import torch as T


z = T.zeros(4, 4)
print(z)
print(z.size())
print(T.numel(z))

运行结果：

创建单位向量-eye()

import torch as T
y = T.eye(3,4)

运行结果: