本文记录了如何在Python中查看Tensor的尺寸和数据类型查看。更新于2019.03.22。
文章目录
Tensor尺寸查看
命令:
x.shape
例子:
input = torch.randn(20,16,50,32)
input.shape
输出为:

注意调用的时候不要用x.shape(),否则会报错:AttributeError: 'Tensor' object has no attribute 'get_shape'
数据类型查看
命令:
x.type()
例子:
input = torch.randn(20,16,50,32)
input.type()
输出:

更多内容,欢迎加入星球讨论。

本文记录了在Python中查看Tensor尺寸和数据类型的方法,介绍了查看尺寸和数据类型的命令、例子及输出,还提醒了调用时的注意事项,最后欢迎读者加入星球讨论更多内容。

368

被折叠的 条评论
为什么被折叠?



