一个例子
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
相关推荐
waterHBO1 小时前
python 爬虫 selenium 笔记编程零零七2 小时前
Python数据分析工具(三)：pymssql的用法AI大模型知识分享2 小时前
Prompt最佳实践｜如何用参考文本让ChatGPT答案更精准？2401_858286113 小时前
52.【C语言】 字符函数和字符串函数(strcat函数)铁松溜达py3 小时前
编译器/工具链环境：GCC vs LLVM/Clang,MSVCRT vs UCRTeveryStudy3 小时前
JavaScript如何判断输入的是空格AIAdvocate4 小时前
Pandas_数据结构详解小言从不摸鱼4 小时前
【AI大模型】ChatGPT模型原理介绍（下）C-SDN花园GGbond5 小时前
【探索数据结构与算法】插入排序：原理、实现与分析（图文详解）迷迭所归处6 小时前
C++ —— 关于vector