tbsm_pytorch.py [561:568]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        batchSize = X[0].shape[0]
        # forward pass
        Z = tbsm(*data_wrap(X,
            lS_o,
            lS_i,
            use_gpu,
            device
        ))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



tbsm_pytorch.py [788:795]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        batchSize = X[0].shape[0]

        Z = tbsm(*data_wrap(X,
            lS_o,
            lS_i,
            use_gpu,
            device
        ))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -