API reference

• torchtt package
  • Submodules
  • torchtt.cpp module
    • cpp_avaible()
  • torchtt.errors module
    • IncompatibleTypes
    • InvalidArguments
    • RankMismatch
    • ShapeMismatch
  • torchtt.grad module
    • grad()
    • grad_list()
    • unwatch()
    • watch()
    • watch_list()
  • torchtt.interpolate module
    • AmenCrossCallbacks
      • AmenCrossCallbacks.compute_enrichment()
      • AmenCrossCallbacks.compute_x_bck()
      • AmenCrossCallbacks.compute_x_fwd()
      • AmenCrossCallbacks.compute_z_bck()
      • AmenCrossCallbacks.compute_z_fwd()
      • AmenCrossCallbacks.update_phis_bck()
      • AmenCrossCallbacks.update_phis_fwd()
    • dmrg_cross()
    • function_interpolate()
  • torchtt.manifold module
    • riemannian_gradient()
    • riemannian_projection()
  • torchtt.nn module
    • AffineTransform
      • AffineTransform.forward()
      • AffineTransform.input_requirement()
    • ComposedTransform
      • ComposedTransform.forward()
      • ComposedTransform.input_requirement()
    • CompressedTTLayer
      • CompressedTTLayer.forward()
    • LinearLayerTT
      • LinearLayerTT.forward()
    • Rank1Shear
      • Rank1Shear.forward()
      • Rank1Shear.input_requirement()
    • SinhArcsinhWarp
      • SinhArcsinhWarp.forward()
      • SinhArcsinhWarp.input_requirement()
    • TTDensityLayer
      • TTDensityLayer.forward()
      • TTDensityLayer.get_tt()
      • TTDensityLayer.input_requireemnt()
      • TTDensityLayer.input_requirement()
      • TTDensityLayer.integration_weights
      • TTDensityLayer.marginalize()
    • Transform
      • Transform.forward()
      • Transform.input_requirement()
    • TriangularPolyShear
      • TriangularPolyShear.forward()
      • TriangularPolyShear.input_requirement()
    • TriangularShear
      • TriangularShear.forward()
      • TriangularShear.input_requirement()
  • torchtt.solvers module
    • amen_solve()
    • cpp_enabled()
  • Module contents
    • TT
      • TT.M
      • TT.N
      • TT.R
      • TT.apply_mask()
      • TT.clone()
      • TT.conj()
      • TT.cpu()
      • TT.cuda()
      • TT.detach()
      • TT.fast_matvec()
      • TT.full()
      • TT.is_cuda()
      • TT.is_ttm
      • TT.mprod()
      • TT.norm()
      • TT.numpy()
      • TT.qtt_to_tens()
      • TT.reduce_dims()
      • TT.round()
      • TT.set_core()
      • TT.sum()
      • TT.t()
      • TT.to()
      • TT.to_qtt()
      • TT.to_ttm()
    • Timer
    • amen_mm()
    • amen_mv()
    • bilinear_form()
    • cat()
    • diag()
    • dmrg_hadamard()
    • dot()
    • elementwise_divide()
    • eye()
    • fast_mm()
    • fast_mv()
    • kron()
    • load()
    • meshgrid()
    • numel()
    • ones()
    • pad()
    • permute()
    • randn()
    • random()
    • rank1TT()
    • reshape()
    • save()
    • shape_mn_to_tuple()
    • shape_tuple_to_mn()
    • zeros()