Pytorch 张量及输入输出计算详细公式
Pytorch 张量及输入输出计算详细公式张量一般文本使用三维张量,适合用于RNN和NLP。如20句话,每句话10个单词,每个单词用100个分量的向量表示,得到的Tensor就是shape=[20, 10, 100]。四维张量,适合用于CNN表示图像。例如100张MNIST数据集的灰度图(通道数为1,如果是RGB图像通道数就是3),每张图高28像素,宽28像素,那么这个Tensor的shape=[100, 1, 28, 28],也就是一个batch的数据维度:[batch_size, channel,
原创
2021-07-22 16:40:51 ·
2238 阅读 ·
0 评论