pytorch张量数据类型

最新推荐文章于 2024-07-19 18:08:20 发布

sci_more

最新推荐文章于 2024-07-19 18:08:20 发布

阅读量48

点赞数

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/weixin_47552564/article/details/132071390

版权

龙良曲P14笔记

import torch
a=torch.randn(2,3)
print(a.type())

a=torch.tensor(1.3) #维度0
b=torch.tensor([1.3]) #维度1
c=torch.randn(2,3) #维度2
d=torch.randn(2,3,1,2) #维度4
print(a,b,c,d)

输出：

torch.FloatTensor
tensor(1.3000) tensor([1.3000]) tensor([[ 0.7555, -1.5726,  0.0640],
        [ 0.0137, -2.0439,  0.3825]]) tensor([[[[-1.7119, -0.6657]],

         [[ 0.4700, -0.5335]],

         [[ 0.5248,  0.6831]]],


        [[[-0.7027, -1.8518]],

         [[ 0.2420, -2.6387]],

         [[ 1.0545, -0.1557]]]])