深度学习框架pytorch中的张量tensor的一些属性介绍

专栏限时打折

正文

创建维度为2的向量

import torch

import numpy as np

a=torch.randn(2,3)

print(a)

print(a.dim())

print(a.shape)

import torch

import numpy as np

a=torch.FloatTensor(2,3)

print(a)

print(a.dim())

print(a.shape)

######################tensor常用的一些属性######################

#tensor的大小

#print(tensor1.shape)#torch.Size([10, 20])

#print(tensor1.size())#torch.Size([10, 20])

#tensor的类型

#print(tensor1.type())#torch.FloatTensor，默认就是这个类型

#tensor的维度

#print(tensor1.dim())#2

# 得到 tensor 的所有元素个数

#print(tensor1.numel())#200