一个例子
python
import torch
dim3=torch.rand(2,3,4)
print(dim3)
print(dim3.shape[0])
print(dim3.shape[1])
print(dim3.shape[2])运行结果
显然shape输出张量对应的维度
关于tensor.shape[0]的理解
weixin_459028002024-03-21 14:25
相关推荐
Sxiaocai1 分钟前
使用 PyTorch 实现并训练 VGGNet 用于 MNIST 分类喵叔哟6 分钟前
重构代码中引入外部方法和引入本地扩展的区别尘浮生12 分钟前
Java项目实战II基于微信小程序的电影院买票选座系统（开发文档+数据库+源码）hopetomorrow26 分钟前
学习路之PHP--使用GROUP BY 发生错误 SELECT list is not in GROUP BY clause .......... 解决小牛itbull35 分钟前
ReactPress vs VuePress vs WordPress请叫我欧皇i44 分钟前
html本地离线引入vant和vue2（详细步骤）nuclear20111 小时前
使用Python 在Excel中创建和取消数据分组 - 详解闲暇部落1 小时前
Kotlin中的?.和!!主要区别GIS瞧葩菜1 小时前
局部修改3dtiles子模型的位置。chnming19871 小时前
STL关联式容器之set