Tensor else data device = data.device resh = torch.reshape(data, [shape[0], ... result.append(torch.reshape(torch.eye(resh.shape[0]).to(device), ...
確定! 回上一頁