擅长:python、mysql、java
<p>要尽可能少地修改示例,请执行以下操作:</p>
<pre><code>import torch
img = torch.rand((3,1080,1080))
RGB_vector = torch.tensor([[[0.2126, 0.7152, 0.0722]]], dtype=img.dtype).permute(2,1,0)
return torch.mean(img * RGB_vector, axis=0).unsqueeze(0)
</code></pre>
<p>这将返回一个(110801080)张量</p>