- 博客(2)
- 收藏
- 关注
RSS订阅原创 pytorch学习二 索引、切片、连接、换位
将输入张量分割成相等形状的chunks（如果可分）。如果沿指定维的张量形状大小不能被split_size 整分， 则最后一个分块会小于其它分块。对于第一行（索引为 0），我们要从 t 的第一行（索引为 0）中收集第 0 列和第 0 列的元素。对于第二行（索引为 1），我们要从 t 的第二行（索引为 1）中收集第 1 列和第 0 列的元素。第一行：t[0, 0] 和 t[0, 0]，即 1 和 1。第二行：t[1, 1] 和 t[1, 0]，即 4 和 3。在给定的维度上对张量序列进行连接操作。
2024-02-28 14:45:24
340
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人