Skip to main content

Module tensor

Module tensor 

Source
Expand description

Metal tensor types for ML operations.

Corresponds to Metal/MTLTensor.hpp.

Tensors represent multi-dimensional arrays for machine learning operations.

Structsยง

Tensor
A multi-dimensional array for machine learning.
TensorDescriptor
Descriptor for creating a tensor.
TensorExtents
Extents (dimensions/strides) for a tensor.