qml.proc¶
This package contains unified functions for frameworkagnostic tensor and array manipulation. Given the input tensorlike object, the call is dispatched to the corresponding array manipulation framework, allowing for endtoend differentiation to be preserved.
Warning
These functions are experimental, and only a subset of common functionality is supported. Furthermore, the names and behaviour of these functions may differ from similar functions in common frameworks; please refer to the function docstrings for more details.
Internally, these functions dispatch by using the TensorBox
class, a container and API for
arraylike objects that allows array manipulation to be performed in a unified manner for supported
tensor/array manipulation frameworks.
The following frameworks are currently supported:
NumPy
Autograd
TensorFlow
PyTorch
Functions¶

Returns the transpose of the tensor by reversing the order of the axes. 

Returns True if two arrays are elementwise equal within a tolerance. 

Returns True if two tensors are elementwise equal along a given axis. 

Casts the given tensor to a new type. 

Casts a tensor to the same dtype as another. 

Convert a tensor to the same type as another. 

Expand the shape of an array by adding a new dimension of size 1 at the specified axis location. 

Returns the name of the package that any array/tensor manipulations will dispatch to. 

Returns a tensor of all ones with the same shape and dtype as the input tensor. 

Returns True if the tensor is considered trainable. 

Returns the shape of the tensor. 

Stack a sequence of tensors along the specified axis. 

Returns the tensor as a NumPy 
Contents
Downloads