Skip to content

Commit 17ef957

Browse files
committed
[mtAE] updated code for newer versions of torch
1 parent a534579 commit 17ef957

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

core/base/mergeTreeAutoencoder/MergeTreeAutoencoder.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -718,7 +718,7 @@ void ttk::MergeTreeAutoencoder::createCustomRecs() {
718718
for(unsigned int j = 0; j < allAlphasInit.size(); ++j) {
719719
allAlphasInit[j]
720720
= torch::randn({layers_[l].getVSTensor().sizes()[1], 1});
721-
auto norm = torch::linalg::vector_norm(
721+
auto norm = torch::linalg_vector_norm(
722722
allAlphasInit[j], 2, 0, false, c10::nullopt);
723723
if(j == 0 or maxNorm.item<float>() < norm.item<float>())
724724
maxNorm = norm;

core/base/mergeTreeNeuralNetwork/MergeTreeNeuralLayer.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,7 @@ void ttk::MergeTreeNeuralLayer::initInputBasisVectors(
403403
torch::Tensor &axes) {
404404
torch::Tensor w = torch::zeros({noVectors, originT.tensor.sizes()[0]});
405405
torch::nn::init::xavier_normal_(w);
406-
axes = torch::linalg::pinv(w);
406+
axes = torch::linalg_pinv(w);
407407
};
408408
initRandomAxes(origin, vSTensor);
409409
if(useGpu_)
@@ -755,7 +755,7 @@ void ttk::MergeTreeNeuralLayer::computeAlphas(
755755
r_data = r_data.cpu();
756756
}
757757
alphasOut
758-
= std::get<0>(torch::linalg::lstsq(r_axes, r_data, c10::nullopt, driver));
758+
= std::get<0>(torch::linalg_lstsq(r_axes, r_data, c10::nullopt, driver));
759759
if(!is_cpu)
760760
alphasOut = alphasOut.to(device);
761761

0 commit comments

Comments
 (0)