This commit is contained in:
Andrei Kislitsyn 2021-03-24 14:00:47 +03:00
parent 078686a046
commit fea53af0ee
2 changed files with 2 additions and 3 deletions

View File

@ -14,7 +14,7 @@ public interface LinearOpsTensorAlgebra<T, TensorType : TensorStructure<T>, Inde
public fun TensorType.qr(): TensorType
//https://pytorch.org/docs/stable/generated/torch.lu.html
public fun TensorType.lu(tol: T): Pair<TensorType, IndexTensorType>
public fun TensorType.lu(): Pair<TensorType, IndexTensorType>
//https://pytorch.org/docs/stable/generated/torch.lu_unpack.html
public fun luPivot(luTensor: TensorType, pivotsTensor: IndexTensorType): Triple<TensorType, TensorType, TensorType>

View File

@ -11,7 +11,7 @@ public class DoubleLinearOpsTensorAlgebra :
TODO("Not yet implemented")
}
override fun DoubleTensor.lu(tol: Double): Pair<DoubleTensor, IntTensor> {
override fun DoubleTensor.lu(): Pair<DoubleTensor, IntTensor> {
checkSquareMatrix(shape)
@ -138,7 +138,6 @@ public class DoubleLinearOpsTensorAlgebra :
TODO("Not yet implemented")
}
override fun DoubleTensor.svd(): Triple<DoubleTensor, DoubleTensor, DoubleTensor> {
TODO("Not yet implemented")
}