Added squeeze
This commit is contained in:
parent
bd3425e7a5
commit
efb23591a9
@ -266,6 +266,10 @@ public open class DoubleTensorAlgebra : TensorPartialDivisionAlgebra<Double, Dou
|
|||||||
TODO("Not yet implemented")
|
TODO("Not yet implemented")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
override fun DoubleTensor.squeeze(dim: Int): DoubleTensor {
|
||||||
|
TODO("Not yet implemented")
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public inline fun <R> DoubleTensorAlgebra(block: DoubleTensorAlgebra.() -> R): R =
|
public inline fun <R> DoubleTensorAlgebra(block: DoubleTensorAlgebra.() -> R): R =
|
||||||
|
@ -86,4 +86,6 @@ public interface TensorAlgebra<T, TensorType : TensorStructure<T>> {
|
|||||||
//https://pytorch.org/docs/stable/generated/torch.flatten.html#torch.flatten
|
//https://pytorch.org/docs/stable/generated/torch.flatten.html#torch.flatten
|
||||||
public fun TensorType.flatten(startDim: Int, endDim: Int): TensorType
|
public fun TensorType.flatten(startDim: Int, endDim: Int): TensorType
|
||||||
|
|
||||||
|
//https://pytorch.org/docs/stable/generated/torch.squeeze.html
|
||||||
|
public fun TensorType.squeeze(dim: Int): TensorType
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user