TensorPlay

Latest development documentation · Updated 2026-09-08. A documentation snapshot for package 1.0.0.dev20260909 is not available.

On this page

Functions 3

#

profile

functionFull reference ↗
tensorplay.cuda.profiler.profile()[source]

Enable profiling.

Context Manager to enabling profile collection by the active profiling tool from CUDA backend. .. rubric:: Example

>>> import tensorplay as tp
>>> model = tp.nn.Linear(20, 30).cuda()
>>> inputs = tp.randn(128, 20).cuda()
>>> with tp.cuda.profiler.profile():
...     model(inputs)
#

start

functionFull reference ↗
tensorplay.cuda.profiler.start()[source]

Starts cuda profiler data collection.

Warning

Raises CudaError in case of it is unable to start the profiler.

#

stop

functionFull reference ↗
tensorplay.cuda.profiler.stop()[source]

Stops cuda profiler data collection.

Warning

Raises CudaError in case of it is unable to stop the profiler.

Search documentation

Search all 1,743 documentation pages.

Keyboard shortcuts

Global

  • /Focus search
  • ?This dialog
  • ,Open settings
  • jAI assistant

Search

  • Navigate results
  • Open result
  • escClose

Package

  • mMain information
  • dDocs
  • .Code
  • -Changelog
  • tTimeline
  • sStats
  • vVersions