Mojo package
tensor
APIs to create and manage tensors in a graph.
Modules
-
io_spec: -
managed_tensor_slice: Implements theManagedTensorSlicetype - a view of a tensor that doesn't own the underlying data. This type is used to build custom graph operations. -
operation_traits: -
tensor_spec: You can import these APIs from themax.tensorpackage. For example: -
transitional: Utilities for transitional period during NDBuffer deprecation.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!