TensorHandle

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

TensorHandle é o tipo usado por operações. Inclui um tipo Scalar , que os internos do compilador podem usar para determinar os tipos de dados dos parâmetros quando eles são extraídos em um programa tensor.

  • Declaração

    public init(_owning cTensorHandle: CTensorHandle)

  • Declaração

    public init(handle: _AnyTensorHandle)

  • Declaração

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

  • Declaração

    public init<C: RandomAccessCollection>(
  _handles: C
) where C.Element: _AnyTensorHandle

  • O número de dimensões do Tensor .

    Declaração

    public var rank: Int { get }

  • A forma do Tensor .

    Declaração

    public var shape: TensorShape { get }

  • O back-end usado para despachar operações.

    Declaração

    public var backend: Device.Backend { get }

  • Crie um ShapedArray com o conteúdo do TensorHandle subjacente. Se o TensorHandle estiver no acelerador, ele será copiado para o host.

    Declaração

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

    Valor de retorno

    Um ShapedArray .

Disponível onde `Scalar`: `TensorFlowScalar`

  • Crie um TensorHandle com um encerramento que inicialize o buffer subjacente.

    scalarsInitializer recebe um buffer com capacidade exatamente suficiente para manter os escalares em um tensor com shape shape . scalarsInitializer deve inicializar todo o buffer, com escalares contíguos em ordem de linha principal.

    Declaração

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