sciai.architecture.FirstOutputCell

class sciai.architecture.FirstOutputCell(backbone)

返回指定网络第一个输出的网络。

参数：

• backbone (Callable) - 原始网络。

输入：

• *inputs (Tensor) - 原始网络输入。

输出：

Union(Tensor, tuple[Tensor])，原始网络的第一个输出。

支持平台：

GPU CPU Ascend

样例：

>>> import mindspore as ms
>>> from mindspore import ops, nn
>>> from sciai.architecture.basic_block import FirstOutputCell
>>> class Net2In3Out(nn.Cell):
>>>     def __init__(self):
>>>         super().__init__()
>>>     def construct(self, x, y):
>>>         out1 = x + y
>>>         out2 = 2 * x + y
>>>         out3 = x * x + 4 * y * y + 3 * y
>>>         return out1.sum(), out2.sum(), out3.sum()
>>> net = Net2In3Out()
>>> first_output_cell = FirstOutputCell(net)
>>> x, y = ops.ones((2, 3), ms.float32), ops.ones((2, 3), ms.float32)
>>> res = first_output_cell(x, y)
>>> print(res)
12.0