特徵矩陣稀疏化(pytorch)

import torchimport   a = torch.tensor([[1, 1, 1, 1.0],im                   [2, 2, 2, 2],di                   [1, 1, 1, 1],return                   [1, 1, 1, 1]])   b, indices = torch.nn.MaxPool2d(2, 2
相關文章
相關標籤/搜索