import torch
a=torch.tensor([[[ 0.0402, -0.1184, -0.7499, -0.3347, -0.7184, -0.7137],
[ 0.0829, -0.3118, -0.2069, -0.8267, -0.6034, -0.1528],
[ 0.2082, -0.1497, -0.3795, -0.2937, -0.5613, -0.0673],
[ 0.3715, -0.089
tensor中[0]进行降维,利用切片的思想
最新推荐文章于 2023-10-20 01:11:07 发布