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")
|
||||
}
|
||||
|
||||
override fun DoubleTensor.squeeze(dim: Int): DoubleTensor {
|
||||
TODO("Not yet implemented")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
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
|
||||
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