当前位置:   article > 正文

关于tensor.shape[0]的理解

关于tensor.shape[0]的理解

一个例子

import torch
dim3=torch.rand(2,3,4)
print(dim3)
print(dim3.shape[0])
print(dim3.shape[1])
print(dim3.shape[2])
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

运行结果
在这里插入图片描述
显然shape输出张量对应的维度

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/262456
推荐阅读
相关标签
  

闽ICP备14008679号