Slicing and Assigning Tensor forge about forge HOT 2 CLOSED

What exactly are you trying to accomplish?

A tensor in Forge is just a placeholder for an MPSImage, and MPSImages have some restrictions on how to use them (to keep them fast). So while you may be tempted to slice a tensor in a tool like TensorFlow, in Forge/MPS there may be a different way to accomplish what you want.

from forge.

Overall, I am trying to use a squeezedet model on iOS. For this question, I am trying to port the convdet layer (post processing of model output) of squeezedet to this iOS example, defined as "_add_interpretation_graph" here: https://github.com/BichenWuUCB/squeezeDet/blob/master/src/nn_skeleton.py#L142

Without all the activations / training code

from forge.