Back to Tensorflow

Shape

tensorflow/lite/g3doc/api_docs/swift/Structs/Tensor/Shape.html

2.21.01.4 KB
Original Source

Shape

public struct Shape : Equatable, Hashable
extension Tensor.Shape: ExpressibleByArrayLiteral

The shape of a Tensor.

`

                rank
                `

The number of dimensions of the Tensor.

Declaration

Swift

public let rank: Int

`

                dimensions
                `

An array of dimensions for the Tensor.

Declaration

Swift

public let dimensions: [Int]

`

                init(_:)
                `

Creates a new instance with the given array of dimensions.

Declaration

Swift

public init(_ dimensions: [Int])

Parameters

| dimensions |

Dimensions for the Tensor.

|

`

                init(_:)
                `

Creates a new instance with the given elements representing the dimensions.

Declaration

Swift

public init(_ elements: Int...)

Parameters

| elements |

Dimensions for the Tensor.

|

`

                init(arrayLiteral:)
                `

Creates a new instance with the given array literal representing the dimensions.

Declaration

Swift

public init(arrayLiteral: Int...)

Parameters

| arrayLiteral |

Dimensions for the Tensor.

|