TensorHandle

public struct TensorHandle<Scalar> where Scalar : _TensorFlowDataTypeCompatible
extension TensorHandle: TensorGroup

TensorHandle est le type utilisé par les opérations. Il inclut un type Scalar , que les internes du compilateur peuvent utiliser pour déterminer les types de données des paramètres lorsqu'ils sont extraits dans un programme tensoriel.

  • Déclaration

    public init(_owning cTensorHandle: CTensorHandle)
  • Déclaration

    public init(handle: _AnyTensorHandle)
  • Déclaration

    public init(_owning tensorHandles: UnsafePointer<CTensorHandle>?)
  • Déclaration

    public init<C: RandomAccessCollection>(
      _handles: C
    ) where C.Element: _AnyTensorHandle
  • Le nombre de dimensions du Tensor .

    Déclaration

    public var rank: Int { get }
  • La forme du Tensor .

    Déclaration

    public var shape: TensorShape { get }
  • Le backend utilisé pour répartir les opérations.

    Déclaration

    public var backend: Device.Backend { get }
  • Créez un ShapedArray avec le contenu du TensorHandle sous-jacent. Si le TensorHandle est sur l'accélérateur, il sera copié sur l'hôte.

    Déclaration

    @inline(never)
    func makeHostCopy() -> ShapedArray<Scalar>

    Valeur de retour

    Un ShapedArray .

Disponible où `Scalar` : `TensorFlowScalar`
  • Créez un TensorHandle avec une fermeture qui initialise le tampon sous-jacent.

    scalarsInitializer reçoit un tampon avec exactement une capacité suffisante pour contenir les scalaires dans un tenseur de shape shape . scalarsInitializer doit initialiser l'intégralité du tampon, avec les scalaires contigus dans l'ordre des lignes principales.

    Déclaration

    public init(
      shape: [Int],
      scalarsInitializer: (UnsafeMutablePointer<Scalar>) -> Void
    )