TorchScript中tensor.view()的替代方案是什么？

我太懂你这糟心的情况了——在TorchScript里想用tensor.view()把bfloat16张量转成int16的视图，结果直接炸出RuntimeError，而且官方文档还明确说了这个view的重载在TorchScript里不支持，搞不好就出未定义行为，确实头疼！

别担心，有个靠谱的替代方法：用torch.bitcast_to()函数。它的作用就是在不改动内存的前提下，把张量按指定的dtype来解析，关键是这个函数是TorchScript完全支持的，不会踩坑。

把你原来代码里的这一行：

t_bits = t.view(int_type)

直接换成：

t_bits = torch.bitcast_to(t, int_type)

这样就能实现和原来view(dtype)一模一样的位视图效果，在TorchScript里跑也完全没问题。

另外还有个备选方案是用torch.Tensor.as_strided()，不过这个方法需要你手动计算形状和步长，用起来繁琐多了，所以优先推荐bitcast_to()。

这里要提个小注意点：bitcast_to()要求输入张量的元素字节数和目标dtype的字节数必须一致。好在你的场景里bfloat16和int16都是2字节，完美匹配，所以不用担心这个问题；如果是其他dtype组合，记得先确认字节数是否对应哦。

备注：内容来源于stack exchange，提问作者Yahor Barkouski