<span class="p">,</span> <span class="n">out_tensor</span><span class="o">=</span><span class="kc">False</span><span class="p">):</span>
<span class="w">        </span><span class="sd">&quot;&quot;&quot;returns prediction probabilities</span>

<span class="sd">        :param input_tensor: test input.</span>
<span class="sd">        :param transform_data: boolean to indicate if data transformation is required.</span>
<span class="sd">        &quot;&quot;&quot;</span>
        <span class="n">input_tensor</spa