Uchwyt Tensora

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

TensorHandle to typ używany przez ops. Zawiera typ Scalar , którego elementy wewnętrzne kompilatora mogą używać do określenia typów danych parametrów, gdy są one wyodrębniane do programu tensorowego.

  • Deklaracja

    public init(_owning cTensorHandle: CTensorHandle)
  • Deklaracja

    public init(handle: _AnyTensorHandle)
  • Deklaracja

    public init(_owning tensorHandles: UnsafePointer<CTensorHandle>?)
  • Deklaracja

    public init<C: RandomAccessCollection>(
      _handles: C
    ) where C.Element: _AnyTensorHandle
  • Liczba wymiarów Tensor .

    Deklaracja

    public var rank: Int { get }
  • Kształt Tensor .

    Deklaracja

    public var shape: TensorShape { get }
  • Backend używany do wysyłania operacji.

    Deklaracja

    public var backend: Device.Backend { get }
  • Utwórz ShapedArray z zawartością bazowego TensorHandle . Jeśli TensorHandle znajduje się w akceleratorze, zostanie skopiowany do hosta.

    Deklaracja

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

    Wartość zwracana

    ShapedArray .

Dostępne, gdy `Scalar`: `TensorFlowScalar`
  • Utwórz TensorHandle z zamknięciem, które inicjuje bazowy bufor.

    scalarsInitializer otrzymuje bufor o pojemności dokładnie wystarczającej do przechowywania skalarów w tensorze o shape . scalarsInitializer musi zainicjować cały bufor z sąsiadującymi skalarami w kolejności od głównego wiersza.

    Deklaracja

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