Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Index Multiprocessing package - tensorplay.multiprocessing Python Module Index Quantization Search tensorplay tensorplay._stax tensorplay.audio tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision

API symbolsNode

latest (dev)

latest (dev)2026-09-08 v1.0.02026-09-06

Copy

Node

class tensorplay.Node

add_post_hook(self: tensorplay._C.Node, hook: collections.abc.Callable[[list[tensorplay._C.TensorBase], list[tensorplay._C.TensorBase]], collections.abc.Sequence[tensorplay._C.TensorBase]]) → None

add_pre_hook(self: tensorplay._C.Node, hook: collections.abc.Callable[[list[tensorplay._C.TensorBase]], collections.abc.Sequence[tensorplay._C.TensorBase]]) → None

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

Unflatten

Unflattens a tensor dim expanding it to a desired shape. For use with Sequential .

tensorplay.nonzero_numpy

Next Page

On this page

Node Node.add_post_hook Node.add_pre_hook