public struct TensorHandle<Scalar> where Scalar : _TensorFlowDataTypeCompatible
extension TensorHandle: TensorGroup
TensorHandle
is the type used by ops. It includes a Scalar
type, which
compiler internals can use to determine the datatypes of parameters when
they are extracted into a tensor program.
-
Declaration
public init(_owning cTensorHandle: CTensorHandle)
-
Declaration
public init(handle: _AnyTensorHandle)
-
Declaration
public init(_owning tensorHandles: UnsafePointer<CTensorHandle>?)
-
Declaration
public init<C: RandomAccessCollection>( _handles: C ) where C.Element: _AnyTensorHandle
-
The number of dimensions of the
Tensor
.Declaration
public var rank: Int { get }
-
The shape of the
Tensor
.Declaration
public var shape: TensorShape { get }
-
The backend used to dispatch ops.
Declaration
public var backend: Device.Backend { get }
-
Create a
ShapedArray
with contents of the underlyingTensorHandle
. If theTensorHandle
is on the accelerator, it will be copied to the host.Declaration
@inline(never) func makeHostCopy() -> ShapedArray<Scalar>
Return Value
A
ShapedArray
.
-
Create a
TensorHandle
with a closure that initializes the underlying buffer.scalarsInitializer
receives a buffer with exactly enough capacity to hold the scalars in a tensor with shapeshape
.scalarsInitializer
must initialize the entire buffer, with contiguous scalars in row-major order.Declaration
public init( shape: [Int], scalarsInitializer: (UnsafeMutablePointer<Scalar>) -> Void )