Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler 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 Upstream anchor labels

tensorplay._shape_funcs.broadcast_shapes tensorplay._shape_funcs.tensordot tensorplay._shape_funcs.unravel_index

API symbols_shape_funcs

Copy

View Markdown Download .md

tensorplay._shape_funcs.unravel_index

tensorplay._shape_funcs.unravel_index(indices, shape)[source]

Converts flat indices into coordinate tuples (one LongTensor per dim).

Indices wrap around modulo the total size, matching torch semantics.

Help improve this page

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

Was this page helpful?

Suggest an edit Report an issue

tensorplay._shape_funcs.tensordot

Contracts input and other over the given dimensions.

tensorplay.adaptive_avg_pool2d

Next Page