欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 新闻 > 焦点 > Pycharm(二十)张量的运算与操作

Pycharm(二十)张量的运算与操作

2025/5/15 13:56:36 来源:https://blog.csdn.net/higerwy/article/details/147839027  浏览:    关键词:Pycharm(二十)张量的运算与操作

一、张量的数据类型转换

 1.演示data.type(trch.DoubleTensor)

#1.创建张量对象  [6 6 6;6 6 6]
data=torch.full([2,3],6)
print(data.dtype)#默认为torch.int64(LongTensor)
#2.转化为double类型
data=data.type(torch.DoubleTensor)
print(data.dtype)
#3.转换成int类型
data=data.type(torch.IntTensor)
print(data.dtype)
#4.转换成Long类型
data=data.type(torch.LongTensor)
print(data)
print(data.dtype)
#5.转换成float类型
data=data.type(torch.FloatTensor)
print(data)
print(data.dtype)

运行结果:

2.演示data.double()

#1.转化为double类型
data1=data.double()
print(data1.dtype)
data2=data.float()
print(data2.dtype)
print(data2.type())
data3=data.short()
print(data3.dtype)

运行结果:

二、张量的类型转换

使用 Tensor.numpy 函数可以将张量转换为 ndarray 数组,但是共享内存,可以使用 copy 函数避免共享

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词