Copy
tensorplay
Tensors
Returns True if obj is a TensorPlay tensor. |
|
Sets the default floating point dtype to |
|
Returns the current default floating point dtype (float32 initially, changed by |
|
Sets the default |
|
Gets the default |
|
Set print options |
Creation Ops
Convert |
|
Indexing, Slicing, Joining, Mutating Ops
Converts flat indices into coordinate tuples (one LongTensor per dim). |
|
Generators
Random sampling
Serialization
Save |
|
Load a checkpoint saved by |
Parallelism
Returns the number of threads used for parallelizing CPU operations |
|
Sets the number of threads used for intraop parallelism on CPU. |
Locally disabling gradient computation
Context-manager that disables gradient calculation. |
|
Context-manager that enables gradient calculation. |
|
Context-manager that sets gradient calculation on or off. |
|
Context manager that enables or disables inference mode. |
Pointwise Ops
Alias of |
|
atan2(y, x) = 2*atan(y / (hypot + x)); native atan2 CPU kernel is currently unregistered in this tree, so compose. |
|
Reduction Ops
Comparison Ops
Spectral Ops
Other Operations
Returns the broadcast shape of the given shapes (right-aligned). |
|
Each row of |
|
Contracts |
|
BLAS and LAPACK Operations
Utilities
Return the dtype to be used for CPU autocasting. |
|
Return the dtype to be used for CUDA autocasting. |
|
Sets whether TensorPlay operations must use "deterministic" algorithms. |
|
Returns True if the global deterministic flag is turned on. |
|
Returns True if the global deterministic flag is set to warn only. |
|
Sets the debug mode for deterministic operations. |
|
Returns the current value of the debug mode for deterministic operations. |
|
Sets the internal precision of float32 matrix multiplications. |
|
Returns the current value of float32 matrix multiplication precision. |
|
String representation of the type of an object. |
Optimizations
Compile a callable through the TensorPlay compiler frontend. |
TensorPlay-specific additions
Members: |
|
Members: |
|
An instance |
|
Sparse/dense storage layout tags (mirrors torch.layout loosely). |
|
Tensor memory layout, mirroring torch.MemoryFormat. |
|
alias of |
|
Instances of |
|
Create a helper decorator for backward methods of custom autograd functions. |
|
Create a helper decorator for |
|
alias of |
|
alias of |
|
Forks the RNG state: code inside the context gets a pristine RNG. |
|
Returns a detailed string describing parallelization settings |
|
Returns the current thread number (starting from 0) in the current parallel region, or 0 in the sequential region |
|
Checks whether the code runs in a parallel region |
|
Summarize a checkpoint without loading tensor payloads. |
|
Help improve this page
Found an error, an unclear step, or a missing example?
