Python PyTorch Tensor.rename用法及代码示例

本文简要介绍python语言中 torch.Tensor.rename 的用法。

用法:

rename(*names, **rename_map)

重命名 self 的维度名称。

主要有两个用途：

self.rename(**rename_map) 返回张量视图，该视图已按照映射 rename_map 中的指定重命名了暗淡。

self.rename(*names) 返回张量视图，使用 names 按位置重命名所有维度。使用 self.rename(None) 在张量上删除名称。

不能同时指定位置参数 names 和关键字参数 rename_map 。

例子：

>>> imgs = torch.rand(2, 3, 5, 7, names=('N', 'C', 'H', 'W'))
>>> renamed_imgs = imgs.rename(N='batch', C='channels')
>>> renamed_imgs.names
('batch', 'channels', 'H', 'W')

>>> renamed_imgs = imgs.rename(None)
>>> renamed_imgs.names
(None,)

>>> renamed_imgs = imgs.rename('batch', 'channel', 'height', 'width')
>>> renamed_imgs.names
('batch', 'channel', 'height', 'width')

警告

命名张量 API 是实验性的，可能会发生变化。