python - 在 Pytorch 中为子张量赋值

Question

t假设我有一个形状为 的 PyTorch 张量(3, 3, 3, 3)，我想为所有和设置t[0, 1, x, y]为。我注意到以下所有语法都有效：zxy

• t[0, 1] = z
• t[0, 1, :] = z
• t[0, 1, :, :] = z
• t[0, 1, ...] = z

在幕后如何执行事物方面有什么不同吗？这些方法中的任何一种是否优于其他方法？

标签： pythonindexingpytorch

Answer 1

:在幕后如何转换为 C++和转换为 C++...存在细微差别：

Python 和 C++ 索引类型的一对一翻译如下：

Python	C++（假设使用命名空间 torch::indexing）
没有任何	没有任何
省略	省略
...	“……”
： 或者 ：：	切片（）或切片（无，无）或切片（无，无，无）

然而，实际上这对性能没有影响。使用在您的上下文中最易读的样式。