happygril3

摘要： https://blog.csdn.net/leo_95/article/details/89946318 matmul 可以进行张量乘法, 输入可以是高维. mat 矩阵对应位相乘，两个tensor输入维度相同，(m×n)和（m×n）,返回（m×n） mm 只能进行矩阵乘法,也就是输入的两个ten 阅读全文

摘要： torch.squeeze() 对数据的维度进行压缩，去掉维数为1的的维度，默认是将a中所有为1的维度删掉。 也可以通过dim指定位置，删掉指定位置的维数为1的维度。torch.unsqueeze()对数据维度进行 import torch x = torch.zeros(3, 2, 4, 1, 2, 1) # dimension of 3*2*4*1*2 print(x.size()) # to 阅读全文