def split(self, split_size, dim=0):
"""Splits this tensor into a tuple of tensors.
See :func:`torch.split`.
"""
return torch.split(self, split_size, dim)
微信
公众号
扫码关注公众号